Weyaxi commited on
Commit
6f46ea5
·
verified ·
1 Parent(s): da8e06f

details edit pray

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -37,7 +37,10 @@ def return_operation_results(from_model, to_model):
37
 
38
 
39
  will_write = json.loads(fs.read_text(path_for))
40
- will_write['config_general']['model_name'] = to_model
 
 
 
41
  will_write = json.dumps(will_write, indent=2)
42
 
43
  liste.extend([CommitOperationAdd(path_in_repo="/".join(remove_from(path_for, from_model, to_model).split("/")[3:]), path_or_fileobj=will_write.encode()),
 
37
 
38
 
39
  will_write = json.loads(fs.read_text(path_for))
40
+ will_write['model_name'] = to_model
41
+ will_write['config']['model_args'] = will_write['config']['model_args'].replace(from_model, to_model)
42
+ will_write['model_name_sanitized'] = to_model.replace("/", "__", 1)
43
+
44
  will_write = json.dumps(will_write, indent=2)
45
 
46
  liste.extend([CommitOperationAdd(path_in_repo="/".join(remove_from(path_for, from_model, to_model).split("/")[3:]), path_or_fileobj=will_write.encode()),