codefuse-ai
/

CodeFuse-CGE-Small

Model card Files Files and versions Community

codefuse-admin commited on Sep 6

Commit

115e007

•

1 Parent(s): a3d3315

Upload README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -43,7 +43,6 @@ from transformers import AutoTokenizer, AutoModel
 model_name_or_path = "CodeFuse-CGE-Small"
 model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, truncation_side='right', padding_side='right')
-model = model.to(torch.bfloat16)
 if torch.cuda.is_available():
     device = 'cuda'

 model_name_or_path = "CodeFuse-CGE-Small"
 model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, truncation_side='right', padding_side='right')
 if torch.cuda.is_available():
     device = 'cuda'