mathewhe
/

DCLM-7B-Chat

Model card Files Files and versions Community

mathewhe commited on Aug 3

Commit

3bf2eac

•

1 Parent(s): 3b9d7a6

Fix typo in argument list

Files changed (1) hide show

chat_class.py +1 -1

chat_class.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 class Chat:
     def __init__(
         self,
-        path="mathewhe/DCLM-7B-Chat"
         device="cuda",
     ):
         r"""

 class Chat:
     def __init__(
         self,
+        path="mathewhe/DCLM-7B-Chat",
         device="cuda",
     ):
         r"""