Commit
•
2360f2a
1
Parent(s):
c0660d2
Update app.py
Browse filestrust_remote_code=True
app.py
CHANGED
|
@@ -102,7 +102,7 @@ CSS = """
|
|
| 102 |
"""
|
| 103 |
|
| 104 |
|
| 105 |
-
tokenizer = AutoTokenizer.from_pretrained("internlm/internlm2_5-7b-chat")
|
| 106 |
model = AutoModelForCausalLM.from_pretrained("internlm/internlm2_5-7b-chat", torch_dtype="auto", device_map="auto",trust_remote_code=True)
|
| 107 |
|
| 108 |
|
|
|
|
| 102 |
"""
|
| 103 |
|
| 104 |
|
| 105 |
+
tokenizer = AutoTokenizer.from_pretrained("internlm/internlm2_5-7b-chat",trust_remote_code=True)
|
| 106 |
model = AutoModelForCausalLM.from_pretrained("internlm/internlm2_5-7b-chat", torch_dtype="auto", device_map="auto",trust_remote_code=True)
|
| 107 |
|
| 108 |
|