From 839e5d9e75e7c5b31c44d355a741022c28086214 Mon Sep 17 00:00:00 2001 From: Yan Gao Date: Thu, 19 Sep 2024 12:34:34 +0100 Subject: [PATCH] refactor(framework) Update dependency and fix a print issue for FlowerTune template (#4220) --- src/py/flwr/cli/new/new.py | 2 +- src/py/flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/py/flwr/cli/new/new.py b/src/py/flwr/cli/new/new.py index d2f7179b45b4..e7dbee894314 100644 --- a/src/py/flwr/cli/new/new.py +++ b/src/py/flwr/cli/new/new.py @@ -275,7 +275,7 @@ def new( ) ) - _add = " huggingface-cli login\n" if framework_str == "flowertune" else "" + _add = " huggingface-cli login\n" if llm_challenge_str else "" print( typer.style( f" cd {package_name}\n" + " pip install -e .\n" + _add + " flwr run\n", diff --git a/src/py/flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl b/src/py/flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl index 8c15739e92f6..8a4d49e7fd84 100644 --- a/src/py/flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl +++ b/src/py/flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl @@ -17,6 +17,7 @@ dependencies = [ "transformers==4.39.3", "sentencepiece==0.2.0", "omegaconf==2.3.0", + "hf_transfer==0.1.8", ] [tool.hatch.build.targets.wheel]