petyussz commited on
Commit
9a755ba
·
verified ·
1 Parent(s): 220bd24

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ pipeline_tag: text-generation
21
  from transformers import AutoModelForCausalLM, AutoTokenizer
22
  import json
23
 
24
- model_name = "petyussz/Qwen2.5-0.5B-Linux-Assistant"
25
 
26
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
27
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
21
  from transformers import AutoModelForCausalLM, AutoTokenizer
22
  import json
23
 
24
+ model_name = "petyussz/shell-assistant-0.5b-v8-it"
25
 
26
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
27
  tokenizer = AutoTokenizer.from_pretrained(model_name)