search

rinongal / textual_inversion

MIT License
2.9k stars 279 forks source link

placeholder_tokens and placeholder_strings #85

Closed tmuskal closed 2 years ago

tmuskal commented 2 years ago

in main.py, the code is referring to the placeholder_tokens param (main is the only place it is referenced), while the model refers to placeholder_strings.

it seem that currently the training process ignores the --placeholder_tokens cmd-line argument.

Am I missing something, or is it simply a typo?

Thanks and keep up the amazing work.

rinongal commented 2 years ago

That's a bug alright. The config file is using placeholder_strings as expected, so you can either fix the parameter name here or modify the placeholders in the config. I'll fix this either way.