Siming Huang commited on
Commit
fcbf7a5
1 Parent(s): 8e06341

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -66,7 +66,7 @@ license_link: https://huggingface.co/infly/OpenCoder-8B-Base/blob/main/LICENSE
66
 
67
  ```python
68
  import torch
69
- from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
70
 
71
  model_name = "infly/OpenCoder-8B-Base"
72
  model = AutoModelForCausalLM.from_pretrained(model_name,
 
66
 
67
  ```python
68
  import torch
69
+ from transformers import AutoTokenizer, AutoModelForCausalLM
70
 
71
  model_name = "infly/OpenCoder-8B-Base"
72
  model = AutoModelForCausalLM.from_pretrained(model_name,