DAMO-NLP
/

SeqGPT-560M

Text Generation

text-generation-inference

Model card Files Files and versions Community

Yirany commited on Sep 1, 2023

Commit

9de1eb8

•

1 Parent(s): 8126bf7

Update README.md

Update model organization

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -35,7 +35,7 @@ The model is fine-tuned based on [BLOOMZ-560M](https://huggingface.co/bigscience
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
-model_name_or_path = 'Yirany/SeqGPT-560M'
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
 model = AutoModelForCausalLM.from_pretrained(model_name_or_path)
 tokenizer.padding_side = 'left'

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
+model_name_or_path = 'DAMO-NLP/SeqGPT-560M'
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
 model = AutoModelForCausalLM.from_pretrained(model_name_or_path)
 tokenizer.padding_side = 'left'