Hi-Satoh commited on
Commit
6c5b23e
·
verified ·
1 Parent(s): 2bcaeeb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -7
README.md CHANGED
@@ -45,19 +45,16 @@ tool use, and recovery from errors.
45
 
46
  ```python
47
  from transformers import AutoModelForCausalLM, AutoTokenizer
48
- from peft import PeftModel
49
  import torch
50
 
51
- base = "Qwen/Qwen3-4B-Instruct-2507"
52
- adapter = "Hi-Satoh/adv_sft_base_merged"
53
 
54
- tokenizer = AutoTokenizer.from_pretrained(base)
55
  model = AutoModelForCausalLM.from_pretrained(
56
- base,
57
  torch_dtype=torch.float16,
58
- device_map="auto",
59
  )
60
- model = PeftModel.from_pretrained(model, adapter)
61
  ```
62
 
63
  ## Sources & Terms (IMPORTANT)
 
45
 
46
  ```python
47
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
48
  import torch
49
 
50
+ model_id = "Hi-Satoh/adv_sft_base_merged"
 
51
 
52
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
53
  model = AutoModelForCausalLM.from_pretrained(
54
+ model_id,
55
  torch_dtype=torch.float16,
56
+ device_map="auto"
57
  )
 
58
  ```
59
 
60
  ## Sources & Terms (IMPORTANT)