manbeast3b
/

deserve_edge4

Model card Files Files and versions

manbeast3b commited on Dec 16, 2024

Commit

4a8eed5

·

verified ·

1 Parent(s): 09d4a2d

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +4 -2

src/pipeline.py CHANGED Viewed

@@ -577,8 +577,10 @@ torch.backends.cuda.matmul.allow_tf32 = True
 torch.backends.cudnn.enabled = True
 torch.backends.cudnn.benchmark = True
-ckpt_id = "black-forest-labs/FLUX.1-schnell"
-ckpt_revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
 def empty_cache():
     gc.collect()
     torch.cuda.empty_cache()

 torch.backends.cudnn.enabled = True
 torch.backends.cudnn.benchmark = True
+# ckpt_id = "black-forest-labs/FLUX.1-schnell"
+# ckpt_revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
+ckpt_id = "RobertML/FLUX.1-schnell-qf8"
+ckpt_revision = "f360ee74b68f38c0b8abd873d0d5800509ed62a2"
 def empty_cache():
     gc.collect()
     torch.cuda.empty_cache()