manbeast3b commited on
Commit
4bbed81
·
verified ·
1 Parent(s): 5309da1

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +2 -2
src/pipeline.py CHANGED
@@ -33,10 +33,10 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
33
  pipeline = compile_pipe(pipeline)
34
  load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
35
 
36
- for _ in range(2):
37
  deepcache_output = pipeline(prompt="polypterid, fattenable, geoparallelotropic, Galeus, galipine, peritoneum, malappropriate, Sekar", output_type="pil", num_inference_steps=20)
38
  pipeline.scheduler.prepare_loss()
39
- for _ in range(4):
40
  pipeline(prompt="polypterid, fattenable, geoparallelotropic, Galeus, galipine, peritoneum, malappropriate, Sekar", output_type="pil", num_inference_steps=20)
41
  return pipeline
42
 
 
33
  pipeline = compile_pipe(pipeline)
34
  load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
35
 
36
+ for _ in range(1):
37
  deepcache_output = pipeline(prompt="polypterid, fattenable, geoparallelotropic, Galeus, galipine, peritoneum, malappropriate, Sekar", output_type="pil", num_inference_steps=20)
38
  pipeline.scheduler.prepare_loss()
39
+ for _ in range(1):
40
  pipeline(prompt="polypterid, fattenable, geoparallelotropic, Galeus, galipine, peritoneum, malappropriate, Sekar", output_type="pil", num_inference_steps=20)
41
  return pipeline
42