Spaces:
Paused
Paused
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,
|