Spaces:

n094t23g
/

self_correcting_text_to_sql_agent

Paused

themissingCRAM commited on 27 days ago

Commit

79dc784

1 Parent(s): 49ab16f

transformerModel

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,9 +121,9 @@ if __name__ == "__main__":
     engine = create_engine("sqlite:///:localhost:")
     engine = init_db(engine)
     model = TransformersModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        device_map="cuda"
-        # model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
     )
     agent = CodeAgent(

     engine = create_engine("sqlite:///:localhost:")
     engine = init_db(engine)
     model = TransformersModel(
+        # model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        device_map="cuda",
+        model_id="meta-llama/Llama-3.2-3B-Instruct"
     )
     agent = CodeAgent(