Update README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,12 @@ license: mit
|
|
| 3 |
tags:
|
| 4 |
- trl
|
| 5 |
- sft
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
---
|
| 7 |
|
| 8 |
# Model Card for Chatbot-Smart-Home
|
|
@@ -96,4 +102,4 @@ inputs = tokenizer(input_text, return_tensors="pt")
|
|
| 96 |
outputs = model.generate(**inputs, max_length=256)
|
| 97 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 98 |
|
| 99 |
-
print(response)
|
|
|
|
| 3 |
tags:
|
| 4 |
- trl
|
| 5 |
- sft
|
| 6 |
+
datasets:
|
| 7 |
+
- yagebin/smart_home_dataset
|
| 8 |
+
language:
|
| 9 |
+
- en
|
| 10 |
+
base_model:
|
| 11 |
+
- mistralai/Mistral-7B-Instruct-v0.3
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for Chatbot-Smart-Home
|
|
|
|
| 102 |
outputs = model.generate(**inputs, max_length=256)
|
| 103 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 104 |
|
| 105 |
+
print(response)
|