Update src/pipeline.py
Browse files- src/pipeline.py +2 -2
src/pipeline.py
CHANGED
|
@@ -32,7 +32,7 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
|
|
| 32 |
|
| 33 |
for _ in range(1):
|
| 34 |
deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|
| 35 |
-
|
| 36 |
for _ in range(2):
|
| 37 |
pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|
| 38 |
return pipeline
|
|
@@ -49,7 +49,7 @@ def infer(request: TextToImageRequest, pipeline: StableDiffusionXLPipeline) -> I
|
|
| 49 |
width=request.width,
|
| 50 |
height=request.height,
|
| 51 |
generator=generator,
|
| 52 |
-
num_inference_steps=
|
| 53 |
cache_interval=1,
|
| 54 |
cache_layer_id=1,
|
| 55 |
cache_block_id=0,
|
|
|
|
| 32 |
|
| 33 |
for _ in range(1):
|
| 34 |
deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|
| 35 |
+
pipeline.scheduler.prepare_loss()
|
| 36 |
for _ in range(2):
|
| 37 |
pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|
| 38 |
return pipeline
|
|
|
|
| 49 |
width=request.width,
|
| 50 |
height=request.height,
|
| 51 |
generator=generator,
|
| 52 |
+
num_inference_steps=13,
|
| 53 |
cache_interval=1,
|
| 54 |
cache_layer_id=1,
|
| 55 |
cache_block_id=0,
|