Spaces:
Runtime error
Runtime error
fix: fixed token handover bug
Browse files- chatmodel.py +1 -0
chatmodel.py
CHANGED
@@ -13,6 +13,7 @@ llama_pipeline = pipeline(
|
|
13 |
model=model,
|
14 |
torch_dtype=torch.float32,
|
15 |
device_map="auto",
|
|
|
16 |
)
|
17 |
|
18 |
# Formatting function for message and history
|
|
|
13 |
model=model,
|
14 |
torch_dtype=torch.float32,
|
15 |
device_map="auto",
|
16 |
+
token = token
|
17 |
)
|
18 |
|
19 |
# Formatting function for message and history
|