Spaces:

NikeZoldyck
/

stable-diff-test

Runtime error

NikeZoldyck commited on Oct 11, 2022

Commit

1796e8a

1 Parent(s): ce9d54d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
-pipe = StableDiffusionPipeline.from_pretrained(model_id, revision="fp16",torch_dtype=torch.float16,use_auth_token=token).to(device)
 def infer(prompt,samples):

 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
+pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16,use_auth_token=token).to(device)
 def infer(prompt,samples):