Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -58,10 +58,12 @@ try:
|
|
58 |
except Exception as e:
|
59 |
logger.warning(f"LoRA loading failed: {e}")
|
60 |
|
61 |
-
# 메모리 최적화
|
62 |
-
|
63 |
-
pipe.
|
64 |
-
|
|
|
|
|
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 |
|