Modify README
Browse files
README.md
CHANGED
@@ -5,7 +5,14 @@ tags:
|
|
5 |
- text-generation
|
6 |
- home-assistant
|
7 |
- voice-assistant
|
|
|
|
|
|
|
8 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
|
|
9 |
---
|
10 |
|
11 |
# 🏠 TinyLLaMA-1.1B Home Assistant Voice Model
|
@@ -34,4 +41,4 @@ query = "turn on the kitchen lights"
|
|
34 |
inputs = tokenizer(query, return_tensors="pt")
|
35 |
outputs = model.generate(**inputs, max_new_tokens=80)
|
36 |
|
37 |
-
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
|
5 |
- text-generation
|
6 |
- home-assistant
|
7 |
- voice-assistant
|
8 |
+
- automation
|
9 |
+
- assistant
|
10 |
+
- home
|
11 |
pipeline_tag: text-generation
|
12 |
+
datasets:
|
13 |
+
- acon96/Home-Assistant-Requests
|
14 |
+
base_model:
|
15 |
+
- TinyLlama/TinyLlama-1.1B-Chat-v1.0
|
16 |
---
|
17 |
|
18 |
# 🏠 TinyLLaMA-1.1B Home Assistant Voice Model
|
|
|
41 |
inputs = tokenizer(query, return_tensors="pt")
|
42 |
outputs = model.generate(**inputs, max_new_tokens=80)
|
43 |
|
44 |
+
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|