Hugging Face's logo

RobbiePasquale
/
lightbulb

Model card Files Community
RobbiePasquale commited on
Commit
d884995
·
verified ·
1 Parent(s): 44f95d3

Update main_menu.py

Browse files
Files changed (1) hide show
  1. main_menu.py +1 -1
main_menu.py CHANGED
@@ -42,7 +42,7 @@ def main():
42
  print("Starting LLM and World Model Training...")
43
  # Directly call the world model main function with appropriate arguments
44
  sys.argv = [
45
- 'lightbulb.py',
46
  '--mode', args.mode,
47
  '--model_name', args.model_name,
48
  '--dataset_name', args.dataset_name,
 
42
  print("Starting LLM and World Model Training...")
43
  # Directly call the world model main function with appropriate arguments
44
  sys.argv = [
45
+ 'lightbulb_custom.py',
46
  '--mode', args.mode,
47
  '--model_name', args.model_name,
48
  '--dataset_name', args.dataset_name,