Wauplin HF staff commited on
Commit
3fd8bf0
1 Parent(s): 64e99f5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,6 @@ from mergekit.options import MergeOptions
14
 
15
  has_gpu = torch.cuda.is_available()
16
 
17
- # Inspired by https://github.com/arcee-ai/mergekit/blob/main/mergekit/scripts/run_yaml.py
18
  merge_options = (
19
  MergeOptions(
20
  copy_tokenizer=True,
@@ -95,6 +94,7 @@ def merge(
95
  config_path = merged_path / "config.yaml"
96
  config_path.write_text(yaml_config)
97
 
 
98
  yield from LogsView.run_thread(
99
  run_merge,
100
  log_level=logging.INFO,
 
14
 
15
  has_gpu = torch.cuda.is_available()
16
 
 
17
  merge_options = (
18
  MergeOptions(
19
  copy_tokenizer=True,
 
94
  config_path = merged_path / "config.yaml"
95
  config_path.write_text(yaml_config)
96
 
97
+ # Taken from https://github.com/arcee-ai/mergekit/blob/main/mergekit/scripts/run_yaml.py
98
  yield from LogsView.run_thread(
99
  run_merge,
100
  log_level=logging.INFO,