Spaces:
Running
on
Zero
Running
on
Zero
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 |
)
|