manbeast3b
/

supatestdiffusion1

Model card Files Files and versions

manbeast3b commited on Nov 19, 2024

Commit

5d16e7c

·

verified ·

1 Parent(s): 73e5b7b

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -48,7 +48,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
     for _ in range(4):
         pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
     # save the compiled pipe
-    save_pipe(pipe, dir="/home/.cache/")
     return pipeline
 def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> Image:

     for _ in range(4):
         pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
     # save the compiled pipe
+    save_pipe(pipeline, dir="/home/.cache/")
     return pipeline
 def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> Image: