codefuse-admin commited on
Commit
115e007
1 Parent(s): a3d3315

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -1
README.md CHANGED
@@ -43,7 +43,6 @@ from transformers import AutoTokenizer, AutoModel
43
  model_name_or_path = "CodeFuse-CGE-Small"
44
  model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
45
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, truncation_side='right', padding_side='right')
46
- model = model.to(torch.bfloat16)
47
 
48
  if torch.cuda.is_available():
49
  device = 'cuda'
 
43
  model_name_or_path = "CodeFuse-CGE-Small"
44
  model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True)
45
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, truncation_side='right', padding_side='right')
 
46
 
47
  if torch.cuda.is_available():
48
  device = 'cuda'