ShaoTengLiu commited on
Commit
ca3f715
1 Parent(s): a53b979
Files changed (1) hide show
  1. 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} {org_dir}'))
52
- return model_dir.as_posix()
 
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(