-
Notifications
You must be signed in to change notification settings - Fork 3k
Fix load config for UIE #4271
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
Fix load config for UIE #4271
Conversation
Thanks for your contribution! |
Codecov Report
@@ Coverage Diff @@
## develop #4271 +/- ##
===========================================
- Coverage 36.35% 36.34% -0.01%
===========================================
Files 419 419
Lines 59169 59172 +3
===========================================
- Hits 21513 21509 -4
- Misses 37656 37663 +7
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
self._check_task_files() | ||
with open(os.path.join(self._task_path, "model_config.json")) as f: | ||
self._init_class = json.load(f)["init_class"] | ||
# TODO: compatible with the model fine-tuned without PretrainedConfig |
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.
这里应该已经向后兼容了吧,对于之前使用model_config.json的模型,这里也能加载进来。所以说TODO已经实现了?
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.
已更新comment
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.
ok
PR types
Others
PR changes
APIs
Description