Skip to content

Support full-precision embeddings in half precision inference mode - #110

Merged
lstein merged 2 commits into
invoke-ai:mainfrom
sajattack:half-precision-embeddings
Aug 28, 2022
Merged

Support full-precision embeddings in half precision inference mode#110
lstein merged 2 commits into
invoke-ai:mainfrom
sajattack:half-precision-embeddings

Conversation

@sajattack

Copy link
Copy Markdown
Contributor

Cast embeddings down to the correct precision

@sajattack sajattack changed the title Support full-precision embeddings in half precision mode Support full-precision embeddings in half precision inference mode Aug 26, 2022
@blessedcoolant

Copy link
Copy Markdown
Collaborator

Interesting. Does it still work with the same memory footprint as the half precision implementation?

@sajattack

sajattack commented Aug 26, 2022

Copy link
Copy Markdown
Contributor Author

Interesting. Does it still work with the same memory footprint as the half precision implementation?

Yes, I use this to run inference with custom embeddings on an 8GB GPU, because it is not enough VRAM for full-precision.

Without this pr implemented, I cannot use custom embeddings, I get an error about a mismatch between half and full precision.

@Oceanswave

Copy link
Copy Markdown
Contributor

lg2m. thanks for the fix!

@lstein

lstein commented Aug 27, 2022

Copy link
Copy Markdown
Collaborator

I'm very grateful for this @sajattack. I'll do a round of testing tomorrow morning. @Oceanswave , could you review as well? (this is an edit- assigned did not mean to assign @blessedcoolant )

@lstein lstein assigned Oceanswave and unassigned blessedcoolant Aug 27, 2022
@morganavr

Copy link
Copy Markdown

Thanks @sajattack for this PR. Now embeddings.pt finally works on my 8GB VRAM GPU!

@lstein
lstein merged commit a7ac93a into invoke-ai:main Aug 28, 2022
austinbrown34 pushed a commit to cognidesign/InvokeAI that referenced this pull request Dec 30, 2022
…ings

Support full-precision embeddings in half precision inference mode
lstein added a commit that referenced this pull request Mar 8, 2026
* Add user management UI for admin and regular users (#106)

* Add user management UI and backend API endpoints

Co-authored-by: lstein <111189+lstein@users.noreply.github.com>

Fix user management feedback: cancel/back navigation, system user filter, tooltip fix

Co-authored-by: lstein <111189+lstein@users.noreply.github.com>

Make Back button on User Management page more prominent

Co-authored-by: lstein <111189+lstein@users.noreply.github.com>

* chore(frontend): typegen

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
Co-authored-by: Lincoln Stein <lincoln.stein@gmail.com>

* Add Confirm Password field to My Profile password change form (#110)

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>

---------

Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
Co-authored-by: Alexander Eichhorn <alex@eichhorn.dev>
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.

5 participants