Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
2c84bf4
1
Parent(s):
e3712a5
update
Browse files- 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 |
|