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.

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

Support any v1.5 checkpoint, various other improvements and fixes... #111

Open
wants to merge 8 commits into
base: main
Choose a base branch
from

Conversation

d8ahazard
Copy link

So, this may be more of a conceptual PR than one that can be directly merged, but here's a list of the changes and modifications I've made:

  1. Support loading weights directly from a v1.5 .ckpt or .safetensors file, falling back to the "stock" weights otherwise.
  2. Add a prompt field. Does it make a difference? IDK. But it's there for funsies.
  3. Add an option to use FP16 or not.
  4. Fix multiple pathing issues when run on Windows due to slashes.
  5. Add an xformerify method to enable xformers on various modules, falling back to attn2.0 if not found.
  6. Update requirements.txt to install CUDA vartiants of Torch, fix other dependencies (Triton is not available on Windows)
  7. Various other generic code cleanup.

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