Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
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('
|
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)
|