Skip to content

Adds support for user-provided safetensor/comfyui checkpoints for Z-Image Turbo#694

Open
davecranwell wants to merge 8 commits intoostris:mainfrom
davecranwell:feature/zit-diffusers-format
Open

Adds support for user-provided safetensor/comfyui checkpoints for Z-Image Turbo#694
davecranwell wants to merge 8 commits intoostris:mainfrom
davecranwell:feature/zit-diffusers-format

Conversation

@davecranwell
Copy link

This PR adds support for locally uploaded *.safetensor files while training on Z-Image Turbo, by converting safetensors files to Diffusers Format during the training job.

I've reused the precedent to use extras_name_or_path to supply the Huggingface repo in cases where the user supplies a local name_or_path pointing to a safetensors file.

The conversion code has been based off notes in another huggingface repo for a converted ZIT model

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant