masanorihirano commited on
Commit
8c894ab
·
verified ·
1 Parent(s): 84e7724

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ Execute the following python code:
45
  import torch
46
  from transformers import AutoTokenizer, AutoModelForCausalLM
47
 
48
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-1.7B", trust_remote_code=True)
49
  model = AutoModelForCausalLM.from_pretrained("pfnet/Qwen3-1.7B-pfn-qfin", device_map="auto", trust_remote_code=True)
50
  text = "日本銀行は"
51
  input_ids = tokenizer(text, return_tensors="pt").input_ids.to(model.device)
 
45
  import torch
46
  from transformers import AutoTokenizer, AutoModelForCausalLM
47
 
48
+ tokenizer = AutoTokenizer.from_pretrained("pfnet/Qwen3-1.7B-pfn-qfin", trust_remote_code=True)
49
  model = AutoModelForCausalLM.from_pretrained("pfnet/Qwen3-1.7B-pfn-qfin", device_map="auto", trust_remote_code=True)
50
  text = "日本銀行は"
51
  input_ids = tokenizer(text, return_tensors="pt").input_ids.to(model.device)