Spaces:
Running
on
Zero
Running
on
Zero
rizavelioglu
commited on
Commit
·
5c1a861
1
Parent(s):
331d5ce
disable TinyAE
Browse files
app.py
CHANGED
|
@@ -56,7 +56,7 @@ class VAETester:
|
|
| 56 |
# "dc-ae-f32c32-sana-1.0": AutoencoderDC.from_pretrained("mit-han-lab/dc-ae-f32c32-sana-1.0-diffusers").to(self.device),
|
| 57 |
"FLUX.1-Kontext": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", subfolder="vae").to(self.device),
|
| 58 |
"FLUX.2": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.2-dev", subfolder="vae").to(self.device),
|
| 59 |
-
"FLUX.2-TinyAutoEncoder": AutoModel.from_pretrained("fal/FLUX.2-Tiny-AutoEncoder", trust_remote_code=True, torch_dtype=torch.bfloat16).to(self.device),
|
| 60 |
}
|
| 61 |
# Define the desired order of models
|
| 62 |
order = [
|
|
@@ -72,7 +72,7 @@ class VAETester:
|
|
| 72 |
# "dc-ae-f32c32-sana-1.0",
|
| 73 |
"FLUX.1-Kontext",
|
| 74 |
"FLUX.2",
|
| 75 |
-
"FLUX.2-TinyAutoEncoder",
|
| 76 |
]
|
| 77 |
|
| 78 |
# Construct the vae_models dictionary in the specified order
|
|
|
|
| 56 |
# "dc-ae-f32c32-sana-1.0": AutoencoderDC.from_pretrained("mit-han-lab/dc-ae-f32c32-sana-1.0-diffusers").to(self.device),
|
| 57 |
"FLUX.1-Kontext": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", subfolder="vae").to(self.device),
|
| 58 |
"FLUX.2": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.2-dev", subfolder="vae").to(self.device),
|
| 59 |
+
# "FLUX.2-TinyAutoEncoder": AutoModel.from_pretrained("fal/FLUX.2-Tiny-AutoEncoder", trust_remote_code=True, torch_dtype=torch.bfloat16).to(self.device),
|
| 60 |
}
|
| 61 |
# Define the desired order of models
|
| 62 |
order = [
|
|
|
|
| 72 |
# "dc-ae-f32c32-sana-1.0",
|
| 73 |
"FLUX.1-Kontext",
|
| 74 |
"FLUX.2",
|
| 75 |
+
# "FLUX.2-TinyAutoEncoder",
|
| 76 |
]
|
| 77 |
|
| 78 |
# Construct the vae_models dictionary in the specified order
|