MDDDDR
/

Llama-3.2-1B-Instruct-FFT-coder-python

Text Generation

text-generation-inference

Model card Files Files and versions

MDDDDR commited on Jan 17

Commit

843c8ab

·

verified ·

1 Parent(s): bc458d2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ datasets:
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-model_id = 'Llama-3.2-1B-Instruct-FFT-coder-python'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id,
                                              device_map="cuda:0",

 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+model_id = 'MDDDDR/Llama-3.2-1B-Instruct-FFT-coder-python'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id,
                                              device_map="cuda:0",