Spaces:

ChenyangSi
/

FreeU

Runtime error

ChenyangSi commited on Sep 28, 2023

Commit

ff69ab2

1 Parent(s): f182225

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -117,7 +117,7 @@ with block:
                 model_id = "CompVis/stable-diffusion-v1-4"
             pip = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-            pip = pip_sd.to("cuda")
             with gr.Row():
                 with gr.Column():

                 model_id = "CompVis/stable-diffusion-v1-4"
             pip = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+            pip = pip.to("cuda")
             with gr.Row():
                 with gr.Column():