1inkusFace commited on
Commit
6a08ac3
·
verified ·
1 Parent(s): 21ebb33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -75,11 +75,11 @@ def upload_to_ftp(filename):
75
 
76
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
77
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
78
- vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
79
 
80
  pipe = StableDiffusion3Pipeline.from_single_file(
81
  "https://huggingface.co/1inkus/sd35-large-UltraReal-bf16-DDUF/blob/main/sd3-bf16-large.dduf",
82
- vae=None,
83
  #tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3"),
84
  use_safetensors=True,
85
  #dtype=torch.bfloat16,
@@ -113,7 +113,7 @@ pipe = StableDiffusion3Pipeline.from_single_file(
113
  #pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
114
 
115
  #pipe.to(device=device) #, dtype=torch.bfloat16)
116
- pipe.vae=vaeX #.to('cpu')
117
 
118
  pipe.to(device)
119
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cuda'))
 
75
 
76
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
77
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
78
+ #vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
79
 
80
  pipe = StableDiffusion3Pipeline.from_single_file(
81
  "https://huggingface.co/1inkus/sd35-large-UltraReal-bf16-DDUF/blob/main/sd3-bf16-large.dduf",
82
+ #vae=None,
83
  #tokenizer_3=T5TokenizerFast.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", add_prefix_space=False, use_fast=True, subfolder="tokenizer_3"),
84
  use_safetensors=True,
85
  #dtype=torch.bfloat16,
 
113
  #pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
114
 
115
  #pipe.to(device=device) #, dtype=torch.bfloat16)
116
+ #pipe.vae=vaeX #.to('cpu')
117
 
118
  pipe.to(device)
119
  upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cuda'))