RobbiePasquale
/

lightbulb

Model card Files Files and versions

RobbiePasquale commited on Oct 11, 2024

Commit

d884995

·

verified ·

1 Parent(s): 44f95d3

Update main_menu.py

Files changed (1) hide show

main_menu.py +1 -1

main_menu.py CHANGED Viewed

@@ -42,7 +42,7 @@ def main():
         print("Starting LLM and World Model Training...")
         # Directly call the world model main function with appropriate arguments
         sys.argv = [
-            'lightbulb.py',
             '--mode', args.mode,
             '--model_name', args.model_name,
             '--dataset_name', args.dataset_name,

         print("Starting LLM and World Model Training...")
         # Directly call the world model main function with appropriate arguments
         sys.argv = [
+            'lightbulb_custom.py',
             '--mode', args.mode,
             '--model_name', args.model_name,
             '--dataset_name', args.dataset_name,