Add abilty to load from safetensors

CarperAI / DRLX

Diffusion Reinforcement Learning Library

MIT License

171 stars 7 forks source link

Add abilty to load from safetensors #16

Closed TingTingin closed 1 year ago

TingTingin commented 1 year ago

It seems that presently you can't load from a safetensors file unless I'm mistaken the library is using diffusers with the sdpipeline under the hood which should support this

shahbuland commented 1 year ago

Solved in #20