kadirnar commited on
Commit
6824a18
1 Parent(s): 29522fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
26
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
27
 
28
  if torch.cuda.is_available():
29
- pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16)
30
 
31
  if ENABLE_CPU_OFFLOAD:
32
  pipe.enable_model_cpu_offload()
 
26
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
27
 
28
  if torch.cuda.is_available():
29
+ pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium", torch_dtype=torch.float16, use_auth_token=huggingface_token)
30
 
31
  if ENABLE_CPU_OFFLOAD:
32
  pipe.enable_model_cpu_offload()