heegyu commited on
Commit
9c12984
1 Parent(s): 65751c6

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ datasets:
4
+ - changpt/ko-lima-vicuna
5
+ - GAIR/lima
6
+ language:
7
+ - en
8
+ - ko
9
+ ---
10
+
11
+ ``` python
12
+ from transformers import pipeline
13
+
14
+ messages = [
15
+ {"role": "user", "content": "Who are you?"},
16
+ ]
17
+ pipe = pipeline("text-generation", model="heegyu/gemma-2-9b-lima", device_map="auto", torch_dtype="auto")
18
+ print(pipe(messages, max_new_tokens=128, eos_token_id=107))
19
+
20
+ ```
21
+
22
+ output: I am an AI assistant, how can I help you today?