Walid-Ahmed commited on
Commit
47613f8
·
verified ·
1 Parent(s): 7493494

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ model_path = snapshot_download(
23
  # Load the Stable Diffusion model
24
  repo = "stabilityai/stable-diffusion-3-medium-diffusers"
25
  #image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None, use_auth_token=huggingface_token)
26
- image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None,torch_dtype=torch.float16)
27
 
28
  #pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
29
 
 
23
  # Load the Stable Diffusion model
24
  repo = "stabilityai/stable-diffusion-3-medium-diffusers"
25
  #image_gen = StableDiffusion3Pipeline.from_pretrained(repo, text_encoder_3=None, tokenizer_3=None, use_auth_token=huggingface_token)
26
+ image_gen = StableDiffusion3Pipeline.from_pretrained(model_path, text_encoder_3=None, tokenizer_3=None,torch_dtype=torch.float16)
27
 
28
  #pipe = StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
29