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

Add utility to inspect a model's dtype/device #14478

Merged
merged 1 commit into from
Dec 31, 2023

Conversation

akx
Copy link
Collaborator

@akx akx commented Dec 30, 2023

Description

Follows up on 3be9074 8100e90: this wraps the logic to retrieve a Torch model's (whether wrapped in a Spandrel descriptor or not) likely dtype and device in a helper function.

Checklist:

@akx akx force-pushed the dtype-inspect branch 2 times, most recently from d0d7184 to 32f8fdc Compare December 30, 2023 22:40
@akx akx marked this pull request as ready for review December 30, 2023 22:48
@akx akx requested a review from AUTOMATIC1111 as a code owner December 30, 2023 22:48
@akx akx marked this pull request as draft December 30, 2023 23:40
@akx

This comment was marked as outdated.

@akx akx marked this pull request as ready for review December 30, 2023 23:50
@AUTOMATIC1111
Copy link
Owner

Can we name the function get_param and just have it return the tensor?

@akx akx marked this pull request as draft December 31, 2023 11:19
@akx akx marked this pull request as ready for review December 31, 2023 11:20
@akx akx marked this pull request as draft December 31, 2023 11:22
@akx akx marked this pull request as ready for review December 31, 2023 14:12
@akx
Copy link
Collaborator Author

akx commented Dec 31, 2023

@AUTOMATIC1111 renamed!

@AUTOMATIC1111 AUTOMATIC1111 merged commit be5f1ac into AUTOMATIC1111:dev Dec 31, 2023
3 checks passed
AUTOMATIC1111 added a commit that referenced this pull request Dec 31, 2023
@w-e-w w-e-w mentioned this pull request Feb 17, 2024
@pawel665j pawel665j mentioned this pull request Apr 16, 2024
ruchej pushed a commit to ruchej/stable-diffusion-webui that referenced this pull request Sep 30, 2024
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.

2 participants