Spaces:

YaYaB
/

text-to-magic

Runtime error

YaYaB commited on Oct 13, 2022

Commit

6ffee60

1 Parent(s): 6b10999

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 dtype = torch.float16 if device == "cuda" else torch.float32
-pipe = StableDiffusionPipeline.from_pretrained("YaYaB/sd-magic-diffusers-test2", torch_dtype=dtype)
 pipe = pipe.to(device)

 context = autocast if device == "cuda" else nullcontext
 dtype = torch.float16 if device == "cuda" else torch.float32
+pipe = StableDiffusionPipeline.from_pretrained("YaYaB/sd-magic-diffusers-test2", revision="5ed9cff0a416a6346aa17aee0d5ed57dfd59b809", torch_dtype=dtype)
 pipe = pipe.to(device)