Spaces:
Paused
Paused
Update app.py
Browse files
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()
|