Spaces:

AlStable
/

AlPrompt

Runtime error

AlStable commited on Jan 24, 2023

Commit

88c8479

1 Parent(s): 32e5089

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ models=[
 def TextToImage(Prompt,model):
   model_id = model
-  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.get_default_dtype)
   pipe = pipe.to("cpu")
   prompt = Prompt
   image = pipe(prompt).images[0]

 def TextToImage(Prompt,model):
   model_id = model
+  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
   pipe = pipe.to("cpu")
   prompt = Prompt
   image = pipe(prompt).images[0]