Small modification to enable usage by external scripts #956
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add parameter inputs to training and argument parsing function to allow this script to be used by an external call.
|
The documentation is not available anymore as the PR was closed or merged. |
patrickvonplaten
approved these changes
Oct 26, 2022
patil-suraj
reviewed
Oct 26, 2022
| @@ -312,8 +316,7 @@ def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: | |||
| return f"{organization}/{model_id}" | |||
|
|
|||
|
|
|||
| def main(): | |||
| args = parse_args() | |||
| def train(args): | |||
There was a problem hiding this comment.
I would say let's not change this
patil-suraj
approved these changes
Oct 26, 2022
|
Hi,
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This change enables dreambooth training to be called by an external script. This makes it much easier to write wrappers around training for GUI's or batch processing scripts. This change does not break existing usage.