-
Notifications
You must be signed in to change notification settings - Fork 881
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
refactor(framework) Update huggingface
template for flwr new
#4169
Conversation
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.
Overall looks fine to me. The only minor comment is in pyproject.toml
, whether we should set prajjwal1/bert-tiny
as default and add a comment for users with access to larger GPUs, e.g.:
model-name = "prajjwal1/bert-tiny" # If you have access to more GPU resources,
# you can try a larger model, such as "distilbert-base-uncased"
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.
Tiny improvements to suppress transformers
warnings.
src/py/flwr/cli/new/templates/app/pyproject.huggingface.toml.tpl
Outdated
Show resolved
Hide resolved
Co-authored-by: Chong Shen Ng <chong.shen@flower.ai>
Co-authored-by: Chong Shen Ng <chong.shen@flower.ai>
Co-authored-by: Chong Shen Ng <chong.shen@flower.ai>
Co-authored-by: Chong Shen Ng <chong.shen@flower.ai> Co-authored-by: Daniel J. Beutel <daniel@flower.ai>
Updates
HF
flwr new
template.needs: #4119