manbeast3b commited on
Commit
25c8545
1 Parent(s): a22cf6c

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. 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(4):
53
- pipeline(prompt="onomancy, aftergo, spirantic, Platyhelmia, modificator, drupaceous, jobbernowl, hereness", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)
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