Yaron Koresh commited on
Commit
9d19d52
·
verified ·
1 Parent(s): f1052d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  if torch.cuda.is_available():
23
  torch.cuda.max_memory_allocated(device=device)
24
- pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True, token=os.getenv('token'))
25
  pipe = pipe.to(device)
26
  else:
27
  pipe = StableDiffusion3Pipeline.from_pretrained(model_id, use_safetensors=True)
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  if torch.cuda.is_available():
23
  torch.cuda.max_memory_allocated(device=device)
24
+ pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True, token=os.getenv('hf_token'))
25
  pipe = pipe.to(device)
26
  else:
27
  pipe = StableDiffusion3Pipeline.from_pretrained(model_id, use_safetensors=True)