Text Generation
Transformers
PyTorch
Chinese
llama
text-generation-inference
Inference Endpoints
fzmnm commited on
Commit
7fc6f58
1 Parent(s): a6f8efa

Update README.md

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