Spaces:

huggingface-projects
/

QR-code-AI-art-generator

Running on Zero

patrickvonplaten commited on Jun 16, 2023

Commit

9fd248b

•

1 Parent(s): be6eaee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,9 +44,9 @@ sd_pipe = StableDiffusionPipeline.from_pretrained(
     torch_dtype=torch.float16,
     safety_checker=None,
 )
 sd_pipe.scheduler = DPMSolverMultistepScheduler.from_config(sd_pipe.scheduler.config)
 sd_pipe.enable_xformers_memory_efficient_attention()
-sd_pipe.enable_model_cpu_offload()
 def resize_for_condition_image(input_image: Image.Image, resolution: int):

     torch_dtype=torch.float16,
     safety_checker=None,
 )
+sd_pipe.to("cuda")
 sd_pipe.scheduler = DPMSolverMultistepScheduler.from_config(sd_pipe.scheduler.config)
 sd_pipe.enable_xformers_memory_efficient_attention()
 def resize_for_condition_image(input_image: Image.Image, resolution: int):