Yaron Koresh commited on
Commit
f043e56
·
verified ·
1 Parent(s): 9f58dda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):