Hugging Face's logo

Spaces:
NikeZoldyck
/
green-screen-composition-transfer
Build error

App Files Community
2
NikeZoldyck commited on
Commit
438d085
•
1 Parent(s): 4650aad

SD diffuser

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
25
  device = "cuda" if torch.cuda.is_available() else "cpu"
26
  context = autocast if device == "cuda" else nullcontext
27
 
28
- pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16,use_auth_token=token).to(device)
29
 
30
 
31
  def infer(prompt,samples):
 
25
  device = "cuda" if torch.cuda.is_available() else "cpu"
26
  context = autocast if device == "cuda" else nullcontext
27
 
28
+ pipe = StableDiffusionPipeline.from_pretrained(model_id,revision="fp16", torch_dtype=torch.float16,use_auth_token=token).to(device)
29
 
30
 
31
  def infer(prompt,samples):