Commit
·
9c6d781
1
Parent(s):
6bdb568
Update README.md
Browse files
README.md
CHANGED
@@ -25,8 +25,8 @@ Model | #Params | d_model | layers | lm loss uniref-100
|
|
25 |
Instantiate a model like so:
|
26 |
|
27 |
from transformers import AutoModel, AutoModelForCausalLM
|
28 |
-
model = AutoModelForCausalLM.from_pretrained("lightonai/
|
29 |
-
tokenizer = AutoTokenizer.from_pretrained("lightonai/
|
30 |
|
31 |
for generation use we support pipelines:
|
32 |
|
|
|
25 |
Instantiate a model like so:
|
26 |
|
27 |
from transformers import AutoModel, AutoModelForCausalLM
|
28 |
+
model = AutoModelForCausalLM.from_pretrained("lightonai/RITA_s, trust_remote_code=True")
|
29 |
+
tokenizer = AutoTokenizer.from_pretrained("lightonai/RITA_s")
|
30 |
|
31 |
for generation use we support pipelines:
|
32 |
|