Update src/pipeline.py
Browse files- src/pipeline.py +2 -2
src/pipeline.py
CHANGED
|
@@ -1127,8 +1127,8 @@ torch.backends.cuda.matmul.allow_tf32 = True
|
|
| 1127 |
torch.backends.cudnn.enabled = True
|
| 1128 |
torch.backends.cudnn.benchmark = True
|
| 1129 |
|
| 1130 |
-
ckpt_id = "
|
| 1131 |
-
ckpt_revision = "
|
| 1132 |
def empty_cache():
|
| 1133 |
gc.collect()
|
| 1134 |
torch.cuda.empty_cache()
|
|
|
|
| 1127 |
torch.backends.cudnn.enabled = True
|
| 1128 |
torch.backends.cudnn.benchmark = True
|
| 1129 |
|
| 1130 |
+
ckpt_id = "RobertML/FLUX.1-schnell-qf8"
|
| 1131 |
+
ckpt_revision = "f360ee74b68f38c0b8abd873d0d5800509ed62a2"
|
| 1132 |
def empty_cache():
|
| 1133 |
gc.collect()
|
| 1134 |
torch.cuda.empty_cache()
|