manbeast3b
/

fire_stang3

Model card Files Files and versions Community

manbeast3b commited on Feb 23

Commit

70e5a8e

·

verified ·

1 Parent(s): bc73a8f

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +8 -1

src/pipeline.py CHANGED Viewed

@@ -53,9 +53,16 @@ def _load_vae_model():
 def _load_transformer_model():
     """Load the transformer model from a specific cached path."""
     transformer_path = os.path.join(
         HF_HUB_CACHE,
-        "models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146"
     )
     return FluxTransformer2DModel.from_pretrained(
         transformer_path,

 def _load_transformer_model():
     """Load the transformer model from a specific cached path."""
+    # transformer_path = os.path.join(
+    #     HF_HUB_CACHE,"models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146",
+    # )
     transformer_path = os.path.join(
         HF_HUB_CACHE,
+        "models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146",
+        "snapshots",
+        model_revision,
+        "transformer"
     )
     return FluxTransformer2DModel.from_pretrained(
         transformer_path,