Update app.py

app.py

@@ -95,7 +95,7 @@ def run_merge_cpu(runner: LogsViewRunner, cli: str, merged_path: str, tmpdirname
 def run_merge_gpu(runner: LogsViewRunner, cli: str, merged_path: str, tmpdirname: str):
     yield from run_merge_cpu(
         runner,
-        cli,
+        cli + " --cuda --low-cpu-memory --read-to-gpu",
         merged_path,
         tmpdirname,
     )
@@ -164,7 +164,7 @@ def merge(yaml_config: str, hf_token: str, repo_name: str, private: bool) -> Ite
     api = huggingface_hub.HfApi(token=hf_token)
     has_gpu = torch.cuda.is_available()
     cli = "mergekit-yaml config.yaml merge --copy-tokenizer --allow-crimes -v" + (
-        " --cuda --low-cpu-memory --read-to-gpu" if (has_gpu) else " --out-shard-size 1B --lazy-unpickle"
+        " --out-shard-size 1B --lazy-unpickle" if (not has_gpu) else ""
     )
 
     with tempfile.TemporaryDirectory(ignore_cleanup_errors=True) as tmpdirname: