Is it possible to save the model as a single safetensor file?
#138
by
kandrupr
- opened
Hello,
I'm trying to load all of the flux components into a FluxPipeline and then save it as a single file. My overall goal is to end up with flux1-schnell.safetensors file.
Attempting to save with FluxPipeline.save_pretrained(...) produces it's individual components.
(base) kandrupr ~/output $ ls
text_encoder text_encoder_2 tokenizer tokenizer_2 transformer vae
(base) kandrupr ~/output $
I was wondering if anyone had some notes on this?