YaoLuzjut commited on
Commit
ade034c
1 Parent(s): 450cc6f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -18,7 +18,7 @@ import torch
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
  # Load the tokenizer and model
21
- model_path = "YaoLuzjut/partial-layer_fine-tuning_Llama-3.1-8B-Instruct"
22
  tokenizer = AutoTokenizer.from_pretrained(model_path)
23
 
24
  device = 'cuda'
 
18
  from transformers import AutoTokenizer, AutoModelForCausalLM
19
 
20
  # Load the tokenizer and model
21
+ model_path = "YaoLuzjut/Llama-3.1-6.3B-It-Alpaca"
22
  tokenizer = AutoTokenizer.from_pretrained(model_path)
23
 
24
  device = 'cuda'