Update test and gitignore

.gitignore

@@ -0,0 +1 @@
+.vscode/launch.json

llamatest.py

@@ -1,7 +1,8 @@
 from transformers import pipeline
 import torch
 import os
-from tot.models import gpt
+from tot.models import gpt24
+from tot.methods.bfs import get_proposals
 
 os.environ["HF_TOKEN"] = os.getenv("HUGGINGTOKEN")
 
@@ -16,7 +17,7 @@ def testThoughtGeneration():
         device_map="auto",
     )
     my_propose_prompt20 = 'Input: 2 8 8 14\nPossible next steps:\n2 + 8 = 10 (left: 8 10 14)\n8 / 2 = 4 (left: 4 8 14)\n14 + 2 = 16 (left: 8 8 16)\n2 * 8 = 16 (left: 8 14 16)\n8 - 2 = 6 (left: 6 8 14)\n14 - 8 = 6 (left: 2 6 8)\n14 / 2 = 7 (left: 7 8 8)\n14 - 2 = 12 (left: 8 8 12)\nInput: 4 5 6 10\nPossible next steps:\n'
-    result = gpt(my_propose_prompt20, model_pipeline, max_tokens=1000, n=1, stop=["Input"])
+    result = gpt24(my_propose_prompt20, model_pipeline, max_tokens=1000, n=1)
     print(result)
 
 testThoughtGeneration()