1inkusFace commited on
Commit
396c2e9
·
verified ·
1 Parent(s): 2f6f1b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,8 +81,8 @@ pipe = StableDiffusion3Pipeline.from_single_file(
81
  "https://huggingface.co/1inkus/sd35-large-UltraReal-bf16-DDUF/blob/main/sd3-bf16-large.dduf",
82
  #tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3"),
83
  use_safetensors=True,
84
- devive_map='auto',
85
- ) #.to(device=device)
86
 
87
  ### pipe = StableDiffusion3Pipeline.from_pretrained(
88
  #"stabilityai # stable-diffusion-3.5-large",
 
81
  "https://huggingface.co/1inkus/sd35-large-UltraReal-bf16-DDUF/blob/main/sd3-bf16-large.dduf",
82
  #tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3"),
83
  use_safetensors=True,
84
+ #devive_map='auto',
85
+ ).to(device=device, dtype=torch.bfloat16)
86
 
87
  ### pipe = StableDiffusion3Pipeline.from_pretrained(
88
  #"stabilityai # stable-diffusion-3.5-large",