Update README.md
Browse files
README.md
CHANGED
@@ -70,5 +70,17 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
|
70 |
|
71 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
72 |
|
73 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
74 |
```
|
|
|
70 |
|
71 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
72 |
|
73 |
+
prompt = """A: γγγ«γ‘γ―
|
74 |
+
B: γγγ«γ‘γ―
|
75 |
+
A: ε₯½γγͺγΉγγΌγγ―δ½γ§γγοΌ
|
76 |
+
B: γ΅γγ«γΌγ§γ
|
77 |
+
A: ε₯½γγͺι£γΉη©γ―δ½γ§γγοΌ
|
78 |
+
B:"""
|
79 |
+
|
80 |
+
pipe(prompt, temperature=0, do_sample=False, return_full_text=False, max_new_tokens=32)
|
81 |
+
# [{"generated_text": "A: black
|
82 |
+
# Q: η·γ―θ±θͺγ§γͺγγ¨γγοΌ
|
83 |
+
# A: green
|
84 |
+
# Q: γͺγ¬γ³γΈγ―θ±θͺγ§γͺγγ¨γγοΌ
|
85 |
+
# A: orange"}]
|
86 |
```
|