Spaces:

mkoot007
/

Text2Image

Runtime error

mkoot007 commited on Oct 24, 2023

Commit

4161e79

•

1 Parent(s): 189f0af

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,6 @@ import torch
 # Load the model and set up the pipeline
 model_id = "prompthero/openjourney"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
 def generate_image(prompt):
  image = pipe(prompt).images[0]

 # Load the model and set up the pipeline
 model_id = "prompthero/openjourney"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 def generate_image(prompt):
  image = pipe(prompt).images[0]