jacky3102 commited on
Commit
00a8485
·
verified ·
1 Parent(s): 91cb770

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -129,7 +129,8 @@ def get_pipeline():
129
  )
130
 
131
  # CPU 优化
132
- pipe.enable_model_cpu_offload() # 逐层卸载到 CPU
 
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") # 注意力切片