Hanlard
/

Pangu_alpha

Text Generation

Model card Files Files and versions Community

Hanlard commited on Feb 21, 2023

Commit

d028b1e

•

1 Parent(s): 84f1acd

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -16,6 +16,9 @@ PanGu-α is proposed by a joint technical team headed by PCNL. It is the first l
 ### Use
 ```python
-    from transformers import AutoTokenizer
     tokenizer = AutoTokenizer.from_pretrained("Hanlard/Pangu_alpha")
 ```

 ### Use
 ```python
+    from transformers import AutoTokenizer, AutoModelForCausalLM
     tokenizer = AutoTokenizer.from_pretrained("Hanlard/Pangu_alpha")
+    model = AutoModelForCausalLM.from_pretrained("imone/pangu_2_6B", trust_remote_code=True)
 ```