Yuto-24 commited on
Commit
a22d444
1 Parent(s): e15785f

Update README.md

Browse files

You can't use pipeline without import. I just add the import package.

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -14,7 +14,7 @@ A100の4枚の環境がおすすめです
14
 
15
  ### Huggingface
16
  ```python
17
- from transformers import AutoTokenizer, AutoModelForCausalLM
18
  import torch
19
 
20
  tokenizer = AutoTokenizer.from_pretrained("lightblue/ao-karasu-72B")
 
14
 
15
  ### Huggingface
16
  ```python
17
+ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
18
  import torch
19
 
20
  tokenizer = AutoTokenizer.from_pretrained("lightblue/ao-karasu-72B")