Spaces:

hxngiee
/

korean-fairy-tales

Runtime error

hxngiee commited on Dec 7, 2023

Commit

b5fc260

1 Parent(s): b31d431

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def get_pipeline(
     pipe = None
     model_id = "./models"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None, requires_safety_checker=False).to("cpu")
     pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
     # if enable_cpu_offload:

     pipe = None
     model_id = "./models"
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32, safety_checker=None, requires_safety_checker=False).to("cpu")
     pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
     # if enable_cpu_offload: