manbeast3b commited on
Commit
d3a201d
·
verified ·
1 Parent(s): 36f8e11

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -55,7 +55,7 @@ def _load_transformer_model():
55
  """Load the transformer model from a specific cached path."""
56
  transformer_path = os.path.join(
57
  HF_HUB_CACHE,
58
- "models--RobertML--FLUX.1-schnell-int8wo/snapshots/307e0777d92df966a3c0f99f31a6ee8957a9857a"
59
  )
60
  return FluxTransformer2DModel.from_pretrained(
61
  transformer_path,
 
55
  """Load the transformer model from a specific cached path."""
56
  transformer_path = os.path.join(
57
  HF_HUB_CACHE,
58
+ "models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146"
59
  )
60
  return FluxTransformer2DModel.from_pretrained(
61
  transformer_path,