manbeast3b
/

GO_SPORTS_

Model card Files Files and versions

manbeast3b commited on Feb 4, 2025

Commit

caf155f

·

verified ·

1 Parent(s): e6e6cf7

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -72,7 +72,7 @@ def load_pipeline() -> Pipeline:
     pipeline.transformer = torch.compile(pipeline.transformer, mode="max-autotune", fullgraph=True)
     # pipeline.vae = torch.compile(pipeline.vae, mode="max-autotune", fullgraph=True)
-    quantize_(pipeline.vae, uintx_weight_only(torch.uint1))
     warmup_ = "controllable varied focus thai warriors entertainment blue golden pink soft tough padthai"
     for _ in range(2):

     pipeline.transformer = torch.compile(pipeline.transformer, mode="max-autotune", fullgraph=True)
     # pipeline.vae = torch.compile(pipeline.vae, mode="max-autotune", fullgraph=True)
+    quantize_(pipeline.vae, float8_weight_only())
     warmup_ = "controllable varied focus thai warriors entertainment blue golden pink soft tough padthai"
     for _ in range(2):