tensorgirl commited on
Commit
4d4493b
·
verified ·
1 Parent(s): d793403

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,10 +28,10 @@ vectordb = Chroma.from_documents(
28
  from langchain_community.llms import HuggingFaceHub
29
 
30
  llm = HuggingFaceHub(
31
- repo_id="google/gemma-2b-it",
32
  task="text-generation",
33
  model_kwargs={
34
- "max_new_tokens": 512,
35
  "top_k": 5,
36
  "temperature": 0.1,
37
  "repetition_penalty": 1.03,
 
28
  from langchain_community.llms import HuggingFaceHub
29
 
30
  llm = HuggingFaceHub(
31
+ repo_id="google/google/gemma-1.1-2b-it",
32
  task="text-generation",
33
  model_kwargs={
34
+ "max_new_tokens": 7000,
35
  "top_k": 5,
36
  "temperature": 0.1,
37
  "repetition_penalty": 1.03,