Spaces:
Configuration error
Configuration error
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 |
|