Update README.md
Browse files
README.md
CHANGED
@@ -31,7 +31,7 @@ KwaiCoder-23BA4-v1 is the latest open-source self-developed code completion mode
|
|
31 |
```python
|
32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
33 |
import torch
|
34 |
-
model_id = "Kwaipilot/KwaiCoder-
|
35 |
tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
|
36 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16,trust_remote_code=True)
|
37 |
text = "#write a quick sort algorithm"
|
@@ -43,7 +43,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True)[len(text):])
|
|
43 |
```python
|
44 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
45 |
import torch
|
46 |
-
model_id = "Kwaipilot/KwaiCoder-
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
|
48 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16,trust_remote_code=True)
|
49 |
text = """<|fim▁begin|>def find_longest_substring(s):
|
@@ -62,7 +62,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True)[len(text):])
|
|
62 |
```
|
63 |
|
64 |
## 3.License
|
65 |
-
This code repository is licensed under the MIT License.
|
66 |
|
67 |
## 4.BibTex
|
68 |
```BibTex
|
|
|
31 |
```python
|
32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
33 |
import torch
|
34 |
+
model_id = "Kwaipilot/KwaiCoder-23B-A4B-v1"
|
35 |
tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
|
36 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16,trust_remote_code=True)
|
37 |
text = "#write a quick sort algorithm"
|
|
|
43 |
```python
|
44 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
45 |
import torch
|
46 |
+
model_id = "Kwaipilot/KwaiCoder-23B-A4B-v1"
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
|
48 |
model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.bfloat16,trust_remote_code=True)
|
49 |
text = """<|fim▁begin|>def find_longest_substring(s):
|
|
|
62 |
```
|
63 |
|
64 |
## 3.License
|
65 |
+
This code repository is licensed under the MIT License.
|
66 |
|
67 |
## 4.BibTex
|
68 |
```BibTex
|