ShaoTengLiu commited on
Commit
2c84bf4
1 Parent(s): e3712a5
Files changed (1) hide show
  1. trainer.py +1 -0
trainer.py CHANGED
@@ -47,6 +47,7 @@ class Trainer:
47
  cwd=org_dir)
48
  except:
49
  temp_path = huggingface_hub.snapshot_download(base_model_id, use_auth_token=token)
 
50
  subprocess.run(shlex.split(f'mv {temp_path} {org_dir}'))
51
  return model_dir.as_posix()
52
 
 
47
  cwd=org_dir)
48
  except:
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