Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ if torch.cuda.is_available():
|
|
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)
|
28 |
pipe = pipe.to(device)
|
29 |
|
30 |
def translate(text,lang):
|
|
|
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, token=os.getenv('hf_token'))
|
28 |
pipe = pipe.to(device)
|
29 |
|
30 |
def translate(text,lang):
|