Arshit Malik commited on
Commit
aeb694b
·
1 Parent(s): c7d537a

fix second syntax error, remove 60s delay

Browse files
Files changed (2) hide show
  1. automation.py +2 -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
- time.sleep(60)
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
- time.sleep(60); continue
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, # CPU mode
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)