Spaces:
Sleeping
Sleeping
Arshit Malik commited on
Commit ·
aeb694b
1
Parent(s): c7d537a
fix second syntax error, remove 60s delay
Browse files- automation.py +2 -2
- kaggle_llm_template.py +1 -1
automation.py
CHANGED
|
@@ -170,7 +170,7 @@ def step_content(run: Path):
|
|
| 170 |
|
| 171 |
log.info(f"Waiting for LLM kernel {info['slug']} ...")
|
| 172 |
log.info(" Sleeping 60s for kernel to register...")
|
| 173 |
-
|
| 174 |
log.info(" (Dataset mount ~0min + inference ~3-5min)")
|
| 175 |
wait_kaggle(info["slug"])
|
| 176 |
|
|
@@ -415,7 +415,7 @@ def main():
|
|
| 415 |
TRIGGER_SKIP_FILE.unlink(missing_ok=True)
|
| 416 |
NEXT_RUN_AFTER = time.time() + 86400
|
| 417 |
log.info("Skipped. Next run in 24h.")
|
| 418 |
-
|
| 419 |
|
| 420 |
if TRIGGER_RUN_FILE.exists():
|
| 421 |
TRIGGER_RUN_FILE.unlink(missing_ok=True)
|
|
|
|
| 170 |
|
| 171 |
log.info(f"Waiting for LLM kernel {info['slug']} ...")
|
| 172 |
log.info(" Sleeping 60s for kernel to register...")
|
| 173 |
+
# time.sleep(60) removed - kernel registers instantly
|
| 174 |
log.info(" (Dataset mount ~0min + inference ~3-5min)")
|
| 175 |
wait_kaggle(info["slug"])
|
| 176 |
|
|
|
|
| 415 |
TRIGGER_SKIP_FILE.unlink(missing_ok=True)
|
| 416 |
NEXT_RUN_AFTER = time.time() + 86400
|
| 417 |
log.info("Skipped. Next run in 24h.")
|
| 418 |
+
# time.sleep(60) removed - kernel registers instantly
|
| 419 |
|
| 420 |
if TRIGGER_RUN_FILE.exists():
|
| 421 |
TRIGGER_RUN_FILE.unlink(missing_ok=True)
|
kaggle_llm_template.py
CHANGED
|
@@ -55,7 +55,7 @@ llm = Llama(
|
|
| 55 |
n_batch=256,
|
| 56 |
n_ubatch=256,
|
| 57 |
flash_attn=False,
|
| 58 |
-
offload_kqv=False,
|
| 59 |
type_k="q4_0",
|
| 60 |
type_v="q4_0",
|
| 61 |
verbose=False)
|
|
|
|
| 55 |
n_batch=256,
|
| 56 |
n_ubatch=256,
|
| 57 |
flash_attn=False,
|
| 58 |
+
offload_kqv=False,
|
| 59 |
type_k="q4_0",
|
| 60 |
type_v="q4_0",
|
| 61 |
verbose=False)
|