Update app.py
Browse files
app.py
CHANGED
|
@@ -129,7 +129,8 @@ def get_pipeline():
|
|
| 129 |
)
|
| 130 |
|
| 131 |
# CPU 优化
|
| 132 |
-
|
|
|
|
| 133 |
pipe.enable_vae_slicing()
|
| 134 |
pipe.enable_vae_tiling()
|
| 135 |
pipe.enable_attention_slicing("auto") # 注意力切片
|
|
|
|
| 129 |
)
|
| 130 |
|
| 131 |
# CPU 优化
|
| 132 |
+
# 注意:enable_model_cpu_offload() 依赖 CUDA(会把模块按需搬到 GPU),
|
| 133 |
+
# 纯 CPU 环境不能用,否则会报 "Found no NVIDIA driver"
|
| 134 |
pipe.enable_vae_slicing()
|
| 135 |
pipe.enable_vae_tiling()
|
| 136 |
pipe.enable_attention_slicing("auto") # 注意力切片
|