Fix typo in argument list
Browse files- chat_class.py +1 -1
chat_class.py
CHANGED
|
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 5 |
class Chat:
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
-
path="mathewhe/DCLM-7B-Chat"
|
| 9 |
device="cuda",
|
| 10 |
):
|
| 11 |
r"""
|
|
|
|
| 5 |
class Chat:
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
+
path="mathewhe/DCLM-7B-Chat",
|
| 9 |
device="cuda",
|
| 10 |
):
|
| 11 |
r"""
|