-
Notifications
You must be signed in to change notification settings - Fork 867
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
Added a function to upload to Huggingface and resume from Huggingface. #348
Conversation
素晴らしい機能追加、ありがとうございます。(私自身では使わないとは思いますが)たいへん有意義かと思います。時間でき次第、レビューいたします。 ところで Thanks for the great work! It will be very useful (although I won't be using it myself). I will review it as soon as I have time. By the way, the diff in |
あ!ほんとですね、自分の環境のフォーマッタを効かせてしまっていました; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
お手数ですが可能ならご対応いただければ幸いです。お忙しいようでしたら私の方でのちほどマージ時に対応します。
I would appreciate it if you could update this if possible. If you are busy, I will update when merging it later.
ご指摘いただいた箇所の修正とtrain_network.py以外の対応を行いました。 I have corrected the points you pointed out and dealt with other than train_network.py. |
修正ありがとうございます。かなり良い感じですね。こちらでテストしてたのですが、最後にモデルをアップロードするときに、以下のエラーが発生してアップロードされないようです(当方Windows環境でテストしています)。恐らく別スレッドでのアップロード中にスクリプトが終了してエラーになっているのかと思いますが、ちょっと解決方法が分かりませんでした。なにかアイデアはありますでしょうか。 Thanks for the update! It looks pretty good. I was testing it in my env, but when uploading the last model, the following error occurs and it doesn't seem to upload (I'm testing in a Windows environment). Perhaps the script terminated during the upload in another thread, causing the error, but I couldn't figure out a solution for a moment. Any ideas?
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
同期呼び出しにすれば問題なく動作するようですが、この方向はいかがでしょうか。
It seems to work fine if we make a synchronous call, how about this idea?
非同期の問題も修正しました。 Also fixed the async issue. You can change to asynchronous processing with |
ありがとうございます! 他のPRも片付きましたので、時間でき次第、確認しマージします。 Thank you very much! I've finished the other PRs and will check and merge them as soon as I have time. |
最終のstateがアップロードされないことにマージ後に気づきましたので、機能追加させていただきました。またデフォルトの可視性をprivateに変更しました。その他、XTIへの追加など細かい修正をしています。お気づきの点があればお教えください。よろしくお願いいたします。 I noticed after merging that the final state isn't uploaded, so I have added the feature. I also changed the default visibility of the repo to private, and added some other minor fixes, including additions to XTI. Please let me know if you notice any issues. Thanks! |
モデルの途中出力、最終出力、stateをhuggingfaceの指定したリポジトリに自動でアップロードする機能を追加しました。
--resume_from_huggingface
についてこれを有効にした場合、
--resume
で指定された情報をもとにhuggingfaceのリポジトリのフォルダからresumeします。--resume
の形式は--resume {repo_id}/{path_in_repo}:{revision}:{repo_type}
になります。例)
--resume_from_huggingface --resume ddpn08/kohya-test/locons/test-locon-000002-state:main:model
Added a function to automatically upload the model's intermediate output, final output, and state to the repository specified by huggingface.
--resume_from_huggingface
If this is enabled, it will resume from the huggingface repository folder based on the information specified with
--resume
.The format of
--resume
is--resume {repo_id}/{path_in_repo}:{revision}:{repo_type}
.ex)
--resume_from_huggingface --resume ddpn08/kohya-test/locons/test-locon-000002-state:main:model