matthewfarant commited on
Commit
546a8cb
1 Parent(s): 3e70d24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -22,12 +22,12 @@ os.environ["GOOGLE_API_KEY"] = os.getenv('GOOGLE_API_KEY')
22
 
23
  # Llama Endpoint
24
  llm = HuggingFaceEndpoint(
25
- repo_id="meta-llama/Meta-Llama-3.1-8B",
26
  task="text-generation",
27
- max_new_tokens = 4000,
28
- do_sample = False,
29
- repetition_penalty = 1.03
30
- )
31
 
32
  llama3 = ChatHuggingFace(llm=llm, temperature = 1)
33
  llama3_json = ChatHuggingFace(llm=llm, format = 'json', temperature = 0)
 
22
 
23
  # Llama Endpoint
24
  llm = HuggingFaceEndpoint(
25
+ repo_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
26
  task="text-generation",
27
+ max_new_tokens=1000,
28
+ do_sample=False,
29
+ repetition_penalty=1.03,
30
+ )
31
 
32
  llama3 = ChatHuggingFace(llm=llm, temperature = 1)
33
  llama3_json = ChatHuggingFace(llm=llm, format = 'json', temperature = 0)