premrajreddy commited on
Commit
463ce72
·
verified ·
1 Parent(s): 900f061

Modify README

Browse files
Files changed (1) hide show
  1. README.md +8 -1
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))