omitakahiro
commited on
Commit
•
2e1ddc1
1
Parent(s):
38705d5
Update README.md
Browse files
README.md
CHANGED
@@ -15,7 +15,7 @@ Stockmark-100b-instruct-v0.1 is an instruction tuned version of [stockmark-100b]
|
|
15 |
```python
|
16 |
import torch
|
17 |
from transformers import AutoTokenizer
|
18 |
-
from peft
|
19 |
|
20 |
prompt_template = """### 指示:
|
21 |
{instruction}
|
|
|
15 |
```python
|
16 |
import torch
|
17 |
from transformers import AutoTokenizer
|
18 |
+
from peft import AutoPeftModelForCausalLM
|
19 |
|
20 |
prompt_template = """### 指示:
|
21 |
{instruction}
|