Spaces:

YaYaB
/

text-to-onepiece

Runtime error

YaYaB commited on Oct 9, 2022

Commit

4116a65

•

1 Parent(s): 0f5c11c

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-onepiece-diffusers-test", 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-onepiece-diffusers-test", revision="92204cfa5e9beedaa97b8a3e2e9eb657e3512722", torch_dtype=dtype)
 pipe = pipe.to(device)