DGSpitzer commited on
Commit
af6862b
1 Parent(s): e81fe92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,8 +21,9 @@ prompt_prefixes = {
21
 
22
  current_model = models[0]
23
 
24
- auth_token = os.environ.get("adminToken") or True
25
- pipe = StableDiffusionPipeline.from_pretrained(current_model, use_auth_token=auth_token, torch_dtype=torchfloat, revision="fp16")
 
26
 
27
  if torch.cuda.is_available():
28
  pipe = pipe.to("cuda")
 
21
 
22
  current_model = models[0]
23
 
24
+ //auth_token = os.environ.get("adminToken") or True
25
+ //pipe = StableDiffusionPipeline.from_pretrained(current_model, use_auth_token=auth_token, torch_dtype=torchfloat, revision="fp16")
26
+ pipe = StableDiffusionPipeline.from_pretrained(current_model, torch_dtype=torchfloat, revision="fp16")
27
 
28
  if torch.cuda.is_available():
29
  pipe = pipe.to("cuda")