Spaces:

AlazarM
/

trenches-train-hezbollah

Paused

AlazarM commited on Mar 8

Commit

063ea42

verified ·

1 Parent(s): 62bbc5a

Deploy training space template for hezbollah

Files changed (2) hide show

__pycache__/run_space.cpython-313.pyc CHANGED Viewed

Binary files a/__pycache__/run_space.cpython-313.pyc and b/__pycache__/run_space.cpython-313.pyc differ

run_space.py CHANGED Viewed

@@ -175,7 +175,7 @@ def train() -> None:
             "--max-completion-length",
             os.environ.get("MAX_COMPLETION_LENGTH", "128"),
             "--vllm-gpu-memory-utilization",
-            os.environ.get("VLLM_GPU_MEMORY_UTILIZATION", "0.35"),
             "--save-strategy",
             os.environ.get("SAVE_STRATEGY", "no"),
             "--output-dir",

             "--max-completion-length",
             os.environ.get("MAX_COMPLETION_LENGTH", "128"),
             "--vllm-gpu-memory-utilization",
+            os.environ.get("VLLM_GPU_MEMORY_UTILIZATION", "0.9"),
             "--save-strategy",
             os.environ.get("SAVE_STRATEGY", "no"),
             "--output-dir",