Update README.md
Browse files
README.md
CHANGED
@@ -33,6 +33,18 @@ warmup_iters=1000;stable_iters=9000
|
|
33 |
A100上7.5小时即可训练完毕。
|
34 |
这个是预训练版本,没有进行偏好微调。
|
35 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
#### 试玩:
|
37 |
```python
|
38 |
temperature=0
|
|
|
33 |
A100上7.5小时即可训练完毕。
|
34 |
这个是预训练版本,没有进行偏好微调。
|
35 |
|
36 |
+
#### 用法:
|
37 |
+
|
38 |
+
```python
|
39 |
+
from transformers import pipeline
|
40 |
+
|
41 |
+
generator = pipeline('text-generation', model='fzmnm/TinyStoriesAdv_92M')
|
42 |
+
story_prompt = "问:什么是鹦鹉?"
|
43 |
+
generated_story = generator(story_prompt, max_length=256)
|
44 |
+
|
45 |
+
print(generated_story[0]['generated_text'])
|
46 |
+
```
|
47 |
+
|
48 |
#### 试玩:
|
49 |
```python
|
50 |
temperature=0
|