Falln87 commited on
Commit
de77766
·
verified ·
1 Parent(s): 44594ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
14
  # GPU support
15
  if torch.cuda.is_available():
16
  torch.cuda.max_memory_allocated(device=device)
17
- pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", token=HF_TOKEN torch_dtype=torch.float16)
18
  pipe.enable_xformers_memory_efficient_attention()
19
  pipe = pipe.to(device)
20
 
 
14
  # GPU support
15
  if torch.cuda.is_available():
16
  torch.cuda.max_memory_allocated(device=device)
17
+ pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", token=HF_TOKEN, torch_dtype=torch.float16)
18
  pipe.enable_xformers_memory_efficient_attention()
19
  pipe = pipe.to(device)
20