Hugging Face's logo

Spaces:
marrtrobinson
/
wan_more_images
Runtime error

App Files Community
kylin0421 commited on
Commit
c1baeff
·
1 Parent(s): 911a56b

cpu offload

Browse files
Files changed (2) hide show
  1. __pycache__/optimization_quantized.cpython-312.pyc +0 -0
  2. optimization_quantized.py +5 -1
__pycache__/optimization_quantized.cpython-312.pyc CHANGED
Binary files a/__pycache__/optimization_quantized.cpython-312.pyc and b/__pycache__/optimization_quantized.cpython-312.pyc differ
 
optimization_quantized.py CHANGED
@@ -192,4 +192,8 @@ def optimize_pipeline_int8(
192
  except AttributeError:
193
  pass
194
 
195
- pipeline.to("cuda")
 
 
 
 
 
192
  except AttributeError:
193
  pass
194
 
195
+ try:
196
+ pipeline.enable_model_cpu_offload()
197
+ except AttributeError:
198
+ # Older diffusers versions might not expose cpu offload helpers.
199
+ pipeline.to("cuda")