hanzla commited on
Commit
1fc2033
·
1 Parent(s): 65721af
Files changed (1) hide show
  1. src/pdfchatbot.py +2 -2
src/pdfchatbot.py CHANGED
@@ -137,8 +137,8 @@ class PDFChatBot:
137
  def create_organic_pipeline(self):
138
  self.pipeline = pipeline(
139
  "text-generation",
140
- model=self.model,
141
- tokenizer=self.tokenizer,
142
  model_kwargs={"torch_dtype": torch.bfloat16},
143
  device="cuda",
144
  )
 
137
  def create_organic_pipeline(self):
138
  self.pipeline = pipeline(
139
  "text-generation",
140
+ model="meta-llama/Meta-Llama-3-8B-Instruct",
141
+ tokenizer="meta-llama/Meta-Llama-3-8B-Instruct",
142
  model_kwargs={"torch_dtype": torch.bfloat16},
143
  device="cuda",
144
  )