rizavelioglu commited on
Commit
5c1a861
·
1 Parent(s): 331d5ce

disable TinyAE

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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