Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -137,7 +137,7 @@ def upload_to_ftp(filename):
|
|
137 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
138 |
|
139 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
140 |
-
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
|
141 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
142 |
#"stabilityai # stable-diffusion-3.5-large",
|
143 |
"ford442/stable-diffusion-3.5-large-bf16",
|
|
|
137 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
138 |
|
139 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
140 |
+
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)
|
141 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
142 |
#"stabilityai # stable-diffusion-3.5-large",
|
143 |
"ford442/stable-diffusion-3.5-large-bf16",
|