Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ pyx = cyper.inline(code, fast_indexing=True, directives=dict(boundscheck=False,
|
|
62 |
|
63 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
64 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
65 |
-
vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, token=True)
|
66 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
67 |
#"stabilityai # stable-diffusion-3.5-large",
|
68 |
"ford442/stable-diffusion-3.5-large-bf16",
|
|
|
62 |
|
63 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
64 |
#vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
|
65 |
+
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)
|
66 |
pipe = StableDiffusion3Pipeline.from_pretrained(
|
67 |
#"stabilityai # stable-diffusion-3.5-large",
|
68 |
"ford442/stable-diffusion-3.5-large-bf16",
|