Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ from datetime import datetime, timezone
|
|
25 |
has_gpu = torch.cuda.is_available()
|
26 |
|
27 |
cli = "mergekit-yaml config.yaml merge --copy-tokenizer" + (
|
28 |
-
" --cuda --low-cpu-memory --allow-crimes" if has_gpu else " --allow-crimes --out-shard-size 1B --lazy-unpickle"
|
29 |
)
|
30 |
|
31 |
MARKDOWN_DESCRIPTION = """
|
|
|
25 |
has_gpu = torch.cuda.is_available()
|
26 |
|
27 |
cli = "mergekit-yaml config.yaml merge --copy-tokenizer" + (
|
28 |
+
" --cuda --low-cpu-memory --allow-crimes --trust-remote-code" if has_gpu else " --allow-crimes --out-shard-size 1B --lazy-unpickle"
|
29 |
)
|
30 |
|
31 |
MARKDOWN_DESCRIPTION = """
|