manbeast3b commited on
Commit
13c97db
·
verified ·
1 Parent(s): e5cff5c

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. 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 = "black-forest-labs/FLUX.1-schnell"
1131
- ckpt_revision = "741f7c3ce8b383c54771c7003378a50191e9efe9"
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()