beomi commited on
Commit
a241282
1 Parent(s): 0c9af5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -36,6 +36,8 @@ pip install -U transformers tokenizers accelerate safetensors
36
  ```python
37
  from transformers import pipeline
38
 
 
 
39
  def gen(x):
40
  a = pipe(
41
  f'### 질문: {x}\n\n### 답변:',
 
36
  ```python
37
  from transformers import pipeline
38
 
39
+ pipe = pipeline('text-generation', model='beomi/KoAlpaca-KoRWKV-6B', device=0)
40
+
41
  def gen(x):
42
  a = pipe(
43
  f'### 질문: {x}\n\n### 답변:',