Taishi-N324 commited on
Commit
e3fcc0e
1 Parent(s): a849304

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -77,6 +77,7 @@ pip install -r requirements.txt
77
 
78
  ```python
79
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
80
 
81
  model_name = "tokyotech-llm/Swallow-MX-8x7b-NVE-v0.1"
82
  tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
 
77
 
78
  ```python
79
  from transformers import AutoModelForCausalLM, AutoTokenizer
80
+ import torch
81
 
82
  model_name = "tokyotech-llm/Swallow-MX-8x7b-NVE-v0.1"
83
  tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")