seawolf2357 commited on
Commit
31a8d15
·
verified ·
1 Parent(s): bfa6fb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -58,10 +58,12 @@ try:
58
  except Exception as e:
59
  logger.warning(f"LoRA loading failed: {e}")
60
 
61
- # 메모리 최적화 활성화
62
- pipe.enable_vae_slicing()
63
- pipe.enable_vae_tiling()
64
- pipe.enable_model_cpu_offload()
 
 
65
 
66
  logger.info("Model loaded and ready")
67
 
 
58
  except Exception as e:
59
  logger.warning(f"LoRA loading failed: {e}")
60
 
61
+ # 메모리 최적화 - WanImageToVideoPipeline에서 지원하는 메서드만 사용
62
+ try:
63
+ pipe.enable_model_cpu_offload()
64
+ logger.info("CPU offload enabled")
65
+ except:
66
+ logger.info("CPU offload not available")
67
 
68
  logger.info("Model loaded and ready")
69