Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
•
ca3f715
1
Parent(s):
a53b979
debug
Browse files- trainer.py +4 -2
trainer.py
CHANGED
@@ -45,11 +45,13 @@ class Trainer:
|
|
45 |
subprocess.run(shlex.split(
|
46 |
f'git clone https://huggingface.co/{base_model_id}'),
|
47 |
cwd=org_dir)
|
|
|
48 |
else:
|
49 |
temp_path = huggingface_hub.snapshot_download(base_model_id, use_auth_token=token)
|
50 |
print(temp_path, org_dir)
|
51 |
-
subprocess.run(shlex.split(f'mv {temp_path} {
|
52 |
-
|
|
|
53 |
|
54 |
def join_model_library_org(self, token: str) -> None:
|
55 |
subprocess.run(
|
|
|
45 |
subprocess.run(shlex.split(
|
46 |
f'git clone https://huggingface.co/{base_model_id}'),
|
47 |
cwd=org_dir)
|
48 |
+
return model_dir.as_posix()
|
49 |
else:
|
50 |
temp_path = huggingface_hub.snapshot_download(base_model_id, use_auth_token=token)
|
51 |
print(temp_path, org_dir)
|
52 |
+
# subprocess.run(shlex.split(f'mv {temp_path} {model_dir.as_posix()}'))
|
53 |
+
# return model_dir.as_posix()
|
54 |
+
return temp_path
|
55 |
|
56 |
def join_model_library_org(self, token: str) -> None:
|
57 |
subprocess.run(
|