Commit 
							
							·
						
						beede7b
	
1
								Parent(s):
							
							db57085
								
Update README_zh.md
Browse files- README_zh.md +2 -2
 
    	
        README_zh.md
    CHANGED
    
    | 
         @@ -45,8 +45,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True) 
     | 
|
| 45 | 
         
             
            model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
         
     | 
| 46 | 
         
             
            model.eval()
         
     | 
| 47 | 
         
             
            model.to(device)
         
     | 
| 48 | 
         
            -
            text = " 
     | 
| 49 | 
         
            -
            tokens = tokenizer.encode_plus(text)['input_ids'] 
     | 
| 50 | 
         
             
            tokens = torch.tensor(tokens)[None,].to(device)
         
     | 
| 51 | 
         
             
            stop_tokens = ["###", "[UNK]", "</s>"]
         
     | 
| 52 | 
         
             
            with torch.no_grad():
         
     | 
| 
         | 
|
| 45 | 
         
             
            model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
         
     | 
| 46 | 
         
             
            model.eval()
         
     | 
| 47 | 
         
             
            model.to(device)
         
     | 
| 48 | 
         
            +
            text = "明天是个"
         
     | 
| 49 | 
         
            +
            tokens = tokenizer.encode_plus(text)['input_ids']
         
     | 
| 50 | 
         
             
            tokens = torch.tensor(tokens)[None,].to(device)
         
     | 
| 51 | 
         
             
            stop_tokens = ["###", "[UNK]", "</s>"]
         
     | 
| 52 | 
         
             
            with torch.no_grad():
         
     |