manbeast3b commited on
Commit
eebbb64
·
verified ·
1 Parent(s): 1262e70

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +7 -0
src/pipeline.py CHANGED
@@ -29,6 +29,13 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
29
 
30
  pipeline.scheduler = SchedulerWrapper(DDIMScheduler.from_config(pipeline.scheduler.config))
31
 
 
 
 
 
 
 
 
32
  pipeline = compile_pipe(pipeline)
33
  load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
34
 
 
29
 
30
  pipeline.scheduler = SchedulerWrapper(DDIMScheduler.from_config(pipeline.scheduler.config))
31
 
32
+ torch._inductor.config.conv_1x1_as_mm = True
33
+ torch._inductor.config.coordinate_descent_tuning = True
34
+ torch._inductor.config.epilogue_fusion = False
35
+ torch._inductor.config.coordinate_descent_check_all_directions = True
36
+ torch._inductor.config.force_fuse_int_mm_with_mul = True
37
+ torch._inductor.config.use_mixed_mm = True
38
+
39
  pipeline = compile_pipe(pipeline)
40
  load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
41