Hugging Face's logo

Spaces:
Nick088
/
stable-diffusion-arena
Running on Zero

App Files Community
1
Nick088 commited on
Commit
266c9b2
·
verified ·
1 Parent(s): c3f5606

Fix another model cpu offload typo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ sd2_1_pipe.enable_model_cpu_offload()
37
  sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
38
  "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
39
  )
40
- sdxl_pipe.enable_cpu_offload()
41
 
42
  # sdxl flash
43
  sdxl_flash_pipe = StableDiffusionXLPipeline.from_pretrained(
 
37
  sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
38
  "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
39
  )
40
+ sdxl_pipe.enable_model_cpu_offload()
41
 
42
  # sdxl flash
43
  sdxl_flash_pipe = StableDiffusionXLPipeline.from_pretrained(