manbeast3b
/

supatest2

Model card Files Files and versions

manbeast3b commited on Nov 20, 2024

Commit

61b487f

·

verified ·

1 Parent(s): 2bb06b3

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -29,7 +29,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
     register_parallel_pipeline_orig(pipeline)
     register_faster_orig_forward(pipeline.unet)
     register_time(pipeline.unet, 0)
-    # pipeline = compile_pipe(pipeline)
     # load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
     np.save('count_array.npy', [0])

     register_parallel_pipeline_orig(pipeline)
     register_faster_orig_forward(pipeline.unet)
     register_time(pipeline.unet, 0)
+    pipeline = compile_pipe(pipeline)
     # load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
     np.save('count_array.npy', [0])