Spaces:

mikegarts
/

lotr

Runtime error

mikegarts commited on Nov 1, 2022

Commit

61d1651

1 Parent(s): 79c9a48

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ READ_TOKEN = os.environ.get('HF_ACCESS_TOKEN', None)
 def get_image_pipe():
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token=READ_TOKEN)
-    pipe.to(pipe.device)
     return pipe
 def get_model():

 def get_image_pipe():
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token=READ_TOKEN)
+    pipe.to('cuda')
     return pipe
 def get_model():