DGSpitzer commited on
Commit
64ed8c9
1 Parent(s): 7e0eccb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ current_model = models[0]
24
 
25
  #auth_token = os.environ.get("test") or True
26
  #pipe = StableDiffusionPipeline.from_pretrained(current_model, use_auth_token=auth_token, torch_dtype=torchfloat, revision="fp16")
27
- pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torchfloat, revision="fp16")
28
 
29
  if torch.cuda.is_available():
30
  pipe = pipe.to("cuda")
 
24
 
25
  #auth_token = os.environ.get("test") or True
26
  #pipe = StableDiffusionPipeline.from_pretrained(current_model, use_auth_token=auth_token, torch_dtype=torchfloat, revision="fp16")
27
+ pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torchfloat)
28
 
29
  if torch.cuda.is_available():
30
  pipe = pipe.to("cuda")