Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

convert SD checkpoints directly from tensor file #337

Closed
ssube opened this issue Apr 15, 2023 · 0 comments
Closed

convert SD checkpoints directly from tensor file #337

ssube opened this issue Apr 15, 2023 · 0 comments
Labels
model/diffusion scope/convert status/fixed issues that have been fixed and released type/feature new features
Milestone
v0.11

Comments

@ssube
Copy link
Owner

ssube commented Apr 15, 2023

The current SD checkpoint conversion code writes the model to .cache/*/working directory in PyTorch format before converting it into ONNX. That is not necessary and diffusers has a helper function to load directly from the .ckpt or .safetensors file: download_from_original_stable_diffusion_ckpt, https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py#L976
@ssube ssube added status/planned issues that have been planned but not started type/feature new features scope/convert model/diffusion labels Apr 15, 2023
@ssube ssube added this to the v0.10 milestone Apr 15, 2023
@ssube ssube modified the milestones: v0.10, v0.11 Apr 24, 2023
@ssube ssube mentioned this issue Apr 29, 2023
Closed
ssube added a commit that referenced this issue Apr 29, 2023
@ssube
fix(api): switch to diffusers ckpt loading, add more pipelines to con…
4c12615 
…version (#337, #356)
@ssube ssube added status/progress issues that are in progress and have a branch status/fixed issues that have been fixed and released and removed status/planned issues that have been planned but not started status/progress issues that are in progress and have a branch labels Apr 29, 2023
@ssube ssube closed this as completed Apr 29, 2023
@ssube ssube mentioned this issue May 1, 2023
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
model/diffusion scope/convert status/fixed issues that have been fixed and released type/feature new features
Projects
None yet
Milestone
v0.11
Development

No branches or pull requests
1 participant
@ssube