MDDDDR commited on
Commit
843c8ab
·
verified ·
1 Parent(s): bc458d2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ datasets:
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
23
  import torch
24
 
25
- model_id = 'Llama-3.2-1B-Instruct-FFT-coder-python'
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(model_id,
28
  device_map="cuda:0",
 
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
23
  import torch
24
 
25
+ model_id = 'MDDDDR/Llama-3.2-1B-Instruct-FFT-coder-python'
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(model_id,
28
  device_map="cuda:0",