themissingCRAM commited on
Commit
e5190f1
·
1 Parent(s): c2cfe36

transformerModel

Browse files
Files changed (3) hide show
  1. README.md +3 -2
  2. app.py +2 -2
  3. requirements.txt +1 -1
README.md CHANGED
@@ -1,13 +1,14 @@
1
  ---
2
- title: bakery_shop_ordering_system_with_recipe_rag
3
  emoji: 🎷
4
  colorFrom: yellow
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 5.23.3
8
  app_file: app.py
9
- pinned: false
10
  short_description: self correcting text to sql agent based on smolagents exampl
 
11
  ---
12
 
13
  bakery shops ordering system with recipe rag
 
1
  ---
2
+ title: bakery_shop_ordering_system_with_recipe_rag
3
  emoji: 🎷
4
  colorFrom: yellow
5
  colorTo: purple
6
  sdk: gradio
7
  sdk_version: 5.23.3
8
  app_file: app.py
9
+ pinned: True
10
  short_description: self correcting text to sql agent based on smolagents exampl
11
+
12
  ---
13
 
14
  bakery shops ordering system with recipe rag
app.py CHANGED
@@ -160,8 +160,8 @@ if __name__ == "__main__":
160
 
161
  # you can use HFApiModel from smolagents if you don't mind paying more money.
162
  model = TransformersModel(
163
- model_id="Qwen/Qwen2.5-Coder-14B-Instruct",
164
- # model_id="meta-llama/Llama-3.2-3B-Instruct",
165
  device_map="cuda"
166
  ,max_new_tokens=5000,torch_dtype=torch.bfloat16
167
  )
 
160
 
161
  # you can use HFApiModel from smolagents if you don't mind paying more money.
162
  model = TransformersModel(
163
+ # model_id="Qwen/Qwen2.5-Coder-14B-Instruct",
164
+ model_id="meta-llama/Llama-3.2-3B-Instruct",
165
  device_map="cuda"
166
  ,max_new_tokens=5000,torch_dtype=torch.bfloat16
167
  )
requirements.txt CHANGED
@@ -11,4 +11,4 @@ langchain_community == 0.3.20
11
  chromadb == 0.6.3
12
  transformers == 4.50.3
13
  accelerate >=0.26.0
14
- torch == 2.6.0
 
11
  chromadb == 0.6.3
12
  transformers == 4.50.3
13
  accelerate >=0.26.0
14
+ torch