csaguiar commited on
Commit
6557a3f
1 Parent(s): a42c7f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def load_translation_models(translation_model_id):
26
 
27
  def pipeline_generate(diffusion_model_id):
28
  pipe = StableDiffusionPipeline.from_pretrained(
29
- diffusion_model_id, torch_dtype=torch.float16, revision="fp16",
30
  use_auth_token=HUGGING_FACE_TOKEN
31
  )
32
  pipe = pipe.to(DEVICE_NAME)
 
26
 
27
  def pipeline_generate(diffusion_model_id):
28
  pipe = StableDiffusionPipeline.from_pretrained(
29
+ diffusion_model_id,
30
  use_auth_token=HUGGING_FACE_TOKEN
31
  )
32
  pipe = pipe.to(DEVICE_NAME)