jokerbit commited on
Commit
a060357
·
verified ·
1 Parent(s): f0c9b97

Add changes

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -0
src/pipeline.py CHANGED
@@ -42,6 +42,7 @@ def load_pipeline() -> Pipeline:
42
  torch_dtype=torch.bfloat16,
43
  ).to("cuda")
44
  quantize_(pipeline.vae, int8_weight_only())
 
45
  pipeline.transformer.to(memory_format=torch.channels_last)
46
 
47
  PROMPT = 'semiconformity, peregrination, quip, twineless, emotionless, tawa, depickle'
 
42
  torch_dtype=torch.bfloat16,
43
  ).to("cuda")
44
  quantize_(pipeline.vae, int8_weight_only())
45
+ torch.compile(pipeline.vae, mode="max-autotune-no-cudagraphs", fullgraph=True)
46
  pipeline.transformer.to(memory_format=torch.channels_last)
47
 
48
  PROMPT = 'semiconformity, peregrination, quip, twineless, emotionless, tawa, depickle'