[chore] Update command to download dataset from huggingface-cli to hf (#3403)
Signed-off-by: Gao Han <hgaoaf@connect.ust.hk>
This commit is contained in:
1 parent
4a24a517ab
commit
54277a8dd0
1 file changed
+2
-2
@@ -53,7 +53,7 @@ def pytest_addoption(parser):
|
||||
"--wan22-i2v-image-source",
|
||||
action="store",
|
||||
default=None,
|
||||
help="Image source for Wan2.2 I2V accuracy tests. Can be local path or remote URL.",
|
||||
help="Image source for Wan2.2 I2V accuracy tests. Can be local path or remote URL",
|
||||
)
|
||||
group.addoption(
|
||||
"--wan22-i2v-online-timeout-seconds",
|
||||
@@ -83,7 +83,7 @@ def _ensure_dataset_snapshot(dataset_id: str) -> Path:
|
||||
return candidates[0]
|
||||
|
||||
subprocess.run(
|
||||
["huggingface-cli", "download", "--repo-type", "dataset", dataset_id],
|
||||
["hf", "download", "--repo-type", "dataset", dataset_id],
|
||||
check=True,
|
||||
)
|
||||
candidates = _dataset_cache_dirs(dataset_id)
|
||||
|
||||
Reference in new issue
Block a user