Skip to content

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

Open
d8ahazard wants to merge 8 commits intomagic-research:mainfrom
d8ahazard:main
Open

Support any v1.5 checkpoint, various other improvements and fixes...#111
d8ahazard wants to merge 8 commits intomagic-research:mainfrom
d8ahazard:main

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