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

CLI option to select model #40

Closed
skorokithakis opened this issue Oct 4, 2022 · 6 comments
Closed

CLI option to select model #40

skorokithakis opened this issue Oct 4, 2022 · 6 comments
Labels
feature request New feature or request

Comments

@skorokithakis
Copy link

I have created a model with Textual Inversion, and I was wondering if there can be an easy CLI option to select the model to generate from?

@brycedrennan
Copy link
Owner

I haven’t played with textual inversion. I assume your model is a small file added to the initial large model. Is that right?

@skorokithakis
Copy link
Author

I think no, it's a 2 GB ckpt file that I think replaces the original model. I'm trying to see if I can add the option myself, but my GPU has very little VRAM so I can either have the windowing system loaded or run the script, which makes development hard...

@skorokithakis
Copy link
Author

Looks like simply changing ckpt_path in load_model_from_config to the path of my model does the trick.

@skorokithakis
Copy link
Author

I don't know how you want to handle passing the option from the CLI into that function, though.

@brycedrennan
Copy link
Owner

Thanks for the idea. I just pushed out 2.3.0 which lets you specify the model via --model-weights-path

@skorokithakis
Copy link
Author

Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants