DanielHesslow commited on
Commit
9c6d781
·
1 Parent(s): 6bdb568

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/RITA_l, trust_remote_code=True")
29
- tokenizer = AutoTokenizer.from_pretrained("lightonai/RITA_l")
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