Update src/pipeline.py
Browse files- src/pipeline.py +4 -2
src/pipeline.py
CHANGED
|
@@ -19,8 +19,10 @@ torch._dynamo.config.suppress_errors = True
|
|
| 19 |
|
| 20 |
Pipeline = None
|
| 21 |
|
| 22 |
-
ckpt_id = "manbeast3b/flux.1-schnell-full1"
|
| 23 |
-
ckpt_revision = "cb1b599b0d712b9aab2c4df3ad27b050a27ec146"
|
|
|
|
|
|
|
| 24 |
def empty_cache():
|
| 25 |
gc.collect()
|
| 26 |
torch.cuda.empty_cache()
|
|
|
|
| 19 |
|
| 20 |
Pipeline = None
|
| 21 |
|
| 22 |
+
# ckpt_id = "manbeast3b/flux.1-schnell-full1"
|
| 23 |
+
# ckpt_revision = "cb1b599b0d712b9aab2c4df3ad27b050a27ec146"
|
| 24 |
+
ckpt_id = "black-forest-labs/FLUX.1-schnell"
|
| 25 |
+
ckpt_revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
|
| 26 |
def empty_cache():
|
| 27 |
gc.collect()
|
| 28 |
torch.cuda.empty_cache()
|