Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ attrdict
|
|
48 |
|
49 |
### QuickStart
|
50 |
|
51 |
-
```
|
52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
53 |
tokenizer = AutoTokenizer.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True)
|
54 |
model = AutoModelForCausalLM.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
|
|
|
48 |
|
49 |
### QuickStart
|
50 |
|
51 |
+
```python
|
52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
53 |
tokenizer = AutoTokenizer.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True)
|
54 |
model = AutoModelForCausalLM.from_pretrained("aigcode/AIGCodeGeek-DS-6.7B", trust_remote_code=True, torch_dtype=torch.bfloat16).cuda()
|