Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -51,7 +51,7 @@ def load_pipeline() -> Pipeline:
|
|
| 51 |
|
| 52 |
pipeline.to(memory_format=torch.channels_last)
|
| 53 |
pipeline.to("cuda")
|
| 54 |
-
pipeline = apply_cache_on_pipe(pipeline,residual_diff_threshold=0.
|
| 55 |
|
| 56 |
for _ in range(3):
|
| 57 |
pipeline(prompt="bangla road is life life is a big party have fun working hard play infinite games win infinite prizes be the worm you dream of", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
|
|
|
|
| 51 |
|
| 52 |
pipeline.to(memory_format=torch.channels_last)
|
| 53 |
pipeline.to("cuda")
|
| 54 |
+
pipeline = apply_cache_on_pipe(pipeline,residual_diff_threshold=0.345)
|
| 55 |
|
| 56 |
for _ in range(3):
|
| 57 |
pipeline(prompt="bangla road is life life is a big party have fun working hard play infinite games win infinite prizes be the worm you dream of", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
|