manbeast3b
/

fire_stang4

Model card Files Files and versions Community

manbeast3b commited on Feb 23

Commit

471b188

·

verified ·

1 Parent(s): be32132

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +2 -2

src/pipeline.py CHANGED Viewed

@@ -46,8 +46,8 @@ def _load_text_encoder_model():
 def _load_vae_model():
     """Load the variational autoencoder (VAE) model with specified configuration."""
     return AutoencoderTiny.from_pretrained(
-        "RobertML/FLUX.1-schnell-vae_e3m2",
-        revision="da0d2cd7815792fb40d084dbd8ed32b63f153d8d",
         torch_dtype=Config.DTYPE
     )

 def _load_vae_model():
     """Load the variational autoencoder (VAE) model with specified configuration."""
     return AutoencoderTiny.from_pretrained(
+        "manbeast3b/FLUX.1-schnell-taef1-float8",
+        revision="7c538d53ec698509788ed88b1305c6bb019bdb4d",
         torch_dtype=Config.DTYPE
     )