leandroaraujodev commited on
Commit
63eec62
·
1 Parent(s): 7bcf569

update settings

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -110,9 +110,9 @@ elif sidebar_option == 'HF Local':
110
  #model_name="HuggingFaceH4/zephyr-7b-beta",
111
  # model_name="meta-llama/Meta-Llama-3-8B",
112
  #model_name="numind/NuExtract-1.5",
113
- model_name="meta-llama/Llama-3.1-8B",
114
 
115
- tokenizer_name="meta-llama/Llama-3.1-8B",
116
  device_map="auto",
117
  # tokenizer_kwargs={"max_length": 2048},
118
  # uncomment this if using CUDA to reduce memory usage
 
110
  #model_name="HuggingFaceH4/zephyr-7b-beta",
111
  # model_name="meta-llama/Meta-Llama-3-8B",
112
  #model_name="numind/NuExtract-1.5",
113
+ model_name="meta-llama/Llama-3.2-3B",
114
 
115
+ tokenizer_name="meta-llama/Llama-3.2-3B",
116
  device_map="auto",
117
  # tokenizer_kwargs={"max_length": 2048},
118
  # uncomment this if using CUDA to reduce memory usage
requirements.txt CHANGED
@@ -12,6 +12,6 @@ openpyxl
12
  llama-index-llms-huggingface
13
  llama-index-llms-huggingface-api
14
  transformers[torch]
15
- huggingface_hub[inference]
16
  llama_index.embeddings.huggingface
17
  llama-index-readers-file
 
12
  llama-index-llms-huggingface
13
  llama-index-llms-huggingface-api
14
  transformers[torch]
15
+ huggingface_hub
16
  llama_index.embeddings.huggingface
17
  llama-index-readers-file