Chandranshu Jain commited on
Commit
d4c2024
·
verified ·
1 Parent(s): 5a7e8e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -87,10 +87,7 @@ def get_conversational_chain():
87
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
88
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
89
 
90
- model_id = "google/gemma-1.1-2b-it"
91
- dtype = torch.bfloat16
92
-
93
- llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
94
 
95
  pt = ChatPromptTemplate.from_template(template)
96
  # Retrieve and generate using the relevant snippets of the blog.
 
87
  #tokenizer = AutoTokenizer.from_pretrained("google/gemma-1.1-2b-it")
88
  #llm = AutoModelForCausalLM.from_pretrained("google/gemma-1.1-2b-it")
89
 
90
+ llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
 
 
 
91
 
92
  pt = ChatPromptTemplate.from_template(template)
93
  # Retrieve and generate using the relevant snippets of the blog.