hmm
Browse files- model/custom_agent.py +2 -0
model/custom_agent.py
CHANGED
|
@@ -22,6 +22,8 @@ Based on https://github.com/huggingface/transformers/blob/main/src/transformers/
|
|
| 22 |
import time
|
| 23 |
import requests
|
| 24 |
from transformers import Agent
|
|
|
|
|
|
|
| 25 |
from utils.logger import log_response
|
| 26 |
|
| 27 |
from transformers import AutoTokenizer
|
|
|
|
| 22 |
import time
|
| 23 |
import requests
|
| 24 |
from transformers import Agent
|
| 25 |
+
from transformers.prompts import CHAT_MESSAGE_PROMPT
|
| 26 |
+
|
| 27 |
from utils.logger import log_response
|
| 28 |
|
| 29 |
from transformers import AutoTokenizer
|