Update src/pipeline.py
Browse files- 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 |
|