Spaces:
Running
on
A100
Running
on
A100
hello
Browse files
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,
|