manbeast3b
/

edgeupdate1

Model card Files Files and versions

manbeast3b commited on Dec 11, 2024

Commit

8152a67

·

verified ·

1 Parent(s): 86e604c

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -96,7 +96,7 @@ def load_pipeline():
     # Transformer model
     transformer_model = FluxTransformer2DModel.from_pretrained(
-        "/root/.cache/huggingface/hub/models--manbeast3b--flux-schnell-transformer2d-int8-mod/snapshots/c911be0ba0d99bb717c242346c21740e7fe20ddf/",
         torch_dtype=dtype,
         use_safetensors=False
     )

     # Transformer model
     transformer_model = FluxTransformer2DModel.from_pretrained(
+        "/home/sandbox/.cache/huggingface/hub/models--manbeast3b--flux-schnell-transformer2d-int8-mod/snapshots/c911be0ba0d99bb717c242346c21740e7fe20ddf/",
         torch_dtype=dtype,
         use_safetensors=False
     )