jokerbit
/

flux-qa-1

Model card Files Files and versions Community

jokerbit commited on Dec 26, 2024

Commit

a237b42

·

verified ·

1 Parent(s): 6142afd

Good

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -49,7 +49,7 @@ def load_pipeline() -> Pipeline:
         CHECKPOINT,
         revision=REVISION,
         transformer=transformer,
-        vae=vae,
         local_files_only=True,
         torch_dtype=torch.bfloat16,
     ).to("cuda")

         CHECKPOINT,
         revision=REVISION,
         transformer=transformer,
+        # vae=vae,
         local_files_only=True,
         torch_dtype=torch.bfloat16,
     ).to("cuda")