update

trainer.py

@@ -41,11 +41,11 @@ class Trainer:
             org_dir = self.checkpoint_dir / org_name
             org_dir.mkdir(exist_ok=True)
         print(f'https://huggingface.co/{base_model_id}')
-        try:
+        if token == None:
             subprocess.run(shlex.split(
                 f'git clone https://huggingface.co/{base_model_id}'),
                             cwd=org_dir)
-        except:
+        else:
             temp_path = huggingface_hub.snapshot_download(base_model_id, use_auth_token=token)
             print(temp_path, org_dir)
             subprocess.run(shlex.split(f'mv {temp_path} {org_dir}'))