manbeast3b
/

testing64

Model card Files Files and versions

manbeast3b commited on Jan 16, 2025

Commit

c305d4f

·

verified ·

1 Parent(s): 0815e5f

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +1 -1

src/pipeline.py CHANGED Viewed

@@ -44,7 +44,7 @@ def load_pipeline() -> Pipeline:
     model = FluxTransformer2DModel.from_pretrained(path, torch_dtype=dtype, use_safetensors=False).to(memory_format=torch.channels_last)
     pipeline = FluxPipeline.from_pretrained(
         ckpt_id,
-        #vae=vae,
         revision=ckpt_revision,
         transformer=model,
         text_encoder_2=text_encoder_2,

     model = FluxTransformer2DModel.from_pretrained(path, torch_dtype=dtype, use_safetensors=False).to(memory_format=torch.channels_last)
     pipeline = FluxPipeline.from_pretrained(
         ckpt_id,
+        vae=vae,
         revision=ckpt_revision,
         transformer=model,
         text_encoder_2=text_encoder_2,