ShaoTengLiu commited on
Commit
d589a23
1 Parent(s): 820086d

update two buttons

Browse files
Files changed (1) hide show
  1. trainer.py +2 -1
trainer.py CHANGED
@@ -39,6 +39,7 @@ class Trainer:
39
  org_name = base_model_id.split('/')[0]
40
  org_dir = self.checkpoint_dir / org_name
41
  org_dir.mkdir(exist_ok=True)
 
42
  subprocess.run(shlex.split(
43
  f'git clone https://huggingface.co/{base_model_id}'),
44
  cwd=org_dir)
@@ -207,7 +208,7 @@ class Trainer:
207
  blend_word_2: str,
208
  eq_params_1: str,
209
  eq_params_2: str,
210
- tuned_model: str = None,
211
  ) -> str:
212
  # if SPACE_ID == ORIGINAL_SPACE_ID:
213
  # raise gr.Error(
 
39
  org_name = base_model_id.split('/')[0]
40
  org_dir = self.checkpoint_dir / org_name
41
  org_dir.mkdir(exist_ok=True)
42
+ print(https://huggingface.co/{base_model_id})
43
  subprocess.run(shlex.split(
44
  f'git clone https://huggingface.co/{base_model_id}'),
45
  cwd=org_dir)
 
208
  blend_word_2: str,
209
  eq_params_1: str,
210
  eq_params_2: str,
211
+ tuned_model: str,
212
  ) -> str:
213
  # if SPACE_ID == ORIGINAL_SPACE_ID:
214
  # raise gr.Error(