codefuse-ai
/

CodeFuse-CGE-Large

Model card Files Files and versions Community

codefuse-admin commited on 25 days ago

Commit

d2b8595

•

1 Parent(s): 8e2d8fc

Upload README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -44,7 +44,6 @@ from transformers import AutoTokenizer, AutoModel
 model_name_or_path = "CodeFuse-CGE-Large"
 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-Large"
 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'