not-lain commited on
Commit
6ddea2a
1 Parent(s): 5623abe

Push model using huggingface_hub.

Browse files
Files changed (1) hide show
  1. README.md +6 -0
README.md CHANGED
@@ -17,6 +17,12 @@ Library: [pxia](https://github.com/not-lain/pxia)
17
  pip install pxia
18
  ```
19
 
 
 
 
 
 
 
20
  ```python
21
  from pxia import GPT2
22
  model = GPT2.from_pretrained("phxia/gpt2")
 
17
  pip install pxia
18
  ```
19
 
20
+ use the AutoModel class
21
+ ```python
22
+ from from pxia AutoModel
23
+ model = AutoModel.from_pretrained("phxia/gpt2")
24
+ ```
25
+ or you can use the model class directly
26
  ```python
27
  from pxia import GPT2
28
  model = GPT2.from_pretrained("phxia/gpt2")