yuvaranianandhan24 commited on
Commit
9f479bd
1 Parent(s): 35c1f9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -13,8 +13,10 @@ load_dotenv()
13
 
14
  # Configure the Llama index settings
15
  Settings.llm = HuggingFaceInferenceAPI(
16
- model_name = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B"),
17
- tokenizer_name = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B"),
 
 
18
  context_window=3900,
19
  token=os.getenv("HF_TOKEN"),
20
  max_new_tokens=1000,
 
13
 
14
  # Configure the Llama index settings
15
  Settings.llm = HuggingFaceInferenceAPI(
16
+ #model_name = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B"),
17
+ #tokenizer_name = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B"),
18
+ tokenizer_name = AutoTokenizer.from_pretrained("gradientai/Llama-3-8B-Instruct-Gradient-1048k"),
19
+ model_name = AutoModelForCausalLM.from_pretrained("gradientai/Llama-3-8B-Instruct-Gradient-1048k"),
20
  context_window=3900,
21
  token=os.getenv("HF_TOKEN"),
22
  max_new_tokens=1000,