huggingface / optimum-intel

🤗 Optimum Intel: Accelerate inference with Intel optimization tools

https://huggingface.co/docs/optimum/main/en/intel/index

Apache License 2.0

409 stars 112 forks source link

add saving safety_checker #990

Closed eaidova closed 1 week ago

eaidova commented 2 weeks ago

What does this PR do?

safety_checker is part of pipeline for sd and sdxl models family. In case, if we provide already converted model and model_index.json contains safety_checker object definition, pipeline loading failed in attempt to load model for it: OSError: Error no file named pytorch_model.bin, model.safetensors, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory

Fixes # (issue)

Before submitting

[ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[ ] Did you make sure to update the documentation with your changes?
[ ] Did you write any new necessary tests?

HuggingFaceDocBuilderDev commented 2 weeks ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.