manbeast3b commited on
Commit
141c336
·
verified ·
1 Parent(s): f8715e2

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +0 -6
src/pipeline.py CHANGED
@@ -53,10 +53,6 @@ def _load_vae_model():
53
 
54
  def _load_transformer_model():
55
  """Load the transformer model from a specific cached path."""
56
- # transformer_path = os.path.join(
57
- # HF_HUB_CACHE,"models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146",
58
-
59
- # )
60
  transformer_path = os.path.join(
61
  HF_HUB_CACHE,
62
  "models--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146",
@@ -136,6 +132,4 @@ def infer(request: TextToImageRequest, pipeline: DiffusionPipeline, generator: G
136
 
137
  # Initialize environment settings when the module is imported
138
  _initialize_environment()
139
-
140
- # For compatibility with other scripts, alias load_pipeline as load
141
  load = load_pipeline
 
53
 
54
  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--manbeast3b--flux.1-schnell-full1/snapshots/cb1b599b0d712b9aab2c4df3ad27b050a27ec146",
 
132
 
133
  # Initialize environment settings when the module is imported
134
  _initialize_environment()
 
 
135
  load = load_pipeline