manbeast3b commited on
Commit
8e509b8
·
verified ·
1 Parent(s): 6f8a753

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -67,7 +67,7 @@ def load_pipeline() -> Pipeline:
67
  )
68
  # pipeline.vae = torch.compile(vae)
69
  pipeline.to("cuda")
70
- pipeline = apply_cache_on_pipe(pipeline,residual_diff_threshold=0.25)
71
  quantize_(pipeline.vae, int4_weight_only())
72
 
73
  warmup_ = "controllable varied focus thai warriors entertainment claude still goat gang gang yeah"
 
67
  )
68
  # pipeline.vae = torch.compile(vae)
69
  pipeline.to("cuda")
70
+ pipeline = apply_cache_on_pipe(pipeline,residual_diff_threshold=0.56)
71
  quantize_(pipeline.vae, int4_weight_only())
72
 
73
  warmup_ = "controllable varied focus thai warriors entertainment claude still goat gang gang yeah"