new code
Browse files- 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=
|
| 141 |
-
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 |
)
|