manbeast3b
commited on
Commit
•
25c8545
1
Parent(s):
a22cf6c
Update src/pipeline.py
Browse files- src/pipeline.py +4 -3
src/pipeline.py
CHANGED
@@ -49,9 +49,10 @@ def load_pipeline() -> Pipeline:
|
|
49 |
pipeline.transformer = torch.compile(pipeline.transformer, mode="reduce-overhead")
|
50 |
quantize_(pipeline.vae, int8_weight_only())
|
51 |
|
52 |
-
for _ in range(
|
53 |
-
pipeline(prompt="
|
54 |
-
|
|
|
55 |
empty_cache()
|
56 |
return pipeline
|
57 |
|
|
|
49 |
pipeline.transformer = torch.compile(pipeline.transformer, mode="reduce-overhead")
|
50 |
quantize_(pipeline.vae, int8_weight_only())
|
51 |
|
52 |
+
for _ in range(3):
|
53 |
+
pipeline(prompt="Platyhelmia, modificator, drupaceous, jobbernowl, hereness", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
|
54 |
+
pipeline(prompt="", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
|
55 |
+
|
56 |
empty_cache()
|
57 |
return pipeline
|
58 |
|