new code
Browse files- src/pdfchatbot.py +0 -1
src/pdfchatbot.py
CHANGED
|
@@ -59,7 +59,6 @@ class PDFChatBot:
|
|
| 59 |
self.pipeline = pipeline(
|
| 60 |
"text-generation",
|
| 61 |
model="meta-llama/Meta-Llama-3-8B-Instruct",
|
| 62 |
-
tokenizer="meta-llama/Meta-Llama-3-8B-Instruct",
|
| 63 |
model_kwargs={"torch_dtype": torch.float16},
|
| 64 |
device="cuda",
|
| 65 |
)
|
|
|
|
| 59 |
self.pipeline = pipeline(
|
| 60 |
"text-generation",
|
| 61 |
model="meta-llama/Meta-Llama-3-8B-Instruct",
|
|
|
|
| 62 |
model_kwargs={"torch_dtype": torch.float16},
|
| 63 |
device="cuda",
|
| 64 |
)
|