Update src/pipeline.py
Browse files- 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
|