Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -97,7 +97,7 @@ class SinaReasonMedicalChat:
|
|
| 97 |
# Add current message
|
| 98 |
messages.append({"role": "user", "content": message})
|
| 99 |
|
| 100 |
-
tokenized = tokenizer.apply_chat_template(messages, return_dict=True)
|
| 101 |
|
| 102 |
input_ids = torch.tensor(tokenized.input_ids, device="cuda").unsqueeze(0)
|
| 103 |
attention_mask = torch.tensor(tokenized.attention_mask, device="cuda").unsqueeze(0)
|
|
|
|
| 97 |
# Add current message
|
| 98 |
messages.append({"role": "user", "content": message})
|
| 99 |
|
| 100 |
+
tokenized = self.tokenizer.apply_chat_template(messages, return_dict=True)
|
| 101 |
|
| 102 |
input_ids = torch.tensor(tokenized.input_ids, device="cuda").unsqueeze(0)
|
| 103 |
attention_mask = torch.tensor(tokenized.attention_mask, device="cuda").unsqueeze(0)
|