Spaces:

Thuchk
/

Stable_diffusion_v1.4

Runtime error

Thuchk commited on May 15, 2023

Commit

d2beeaa

•

1 Parent(s): 17933fc

fix cuda unavailable

Files changed (1) hide show

gradio_app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 model_path = "sd-pokemon-model"
 pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16, low_cpu_mem_usage=False, device_map=None)
-pipe.to("cuda")
 def predict(prompt):
     image = pipe(prompt=prompt).images[0]

 model_path = "sd-pokemon-model"
 pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16, low_cpu_mem_usage=False, device_map=None)
+# pipe.to("cuda")
 def predict(prompt):
     image = pipe(prompt=prompt).images[0]