Views
No views yet
python -m pip install trl1from transformers import pipeline
2
3generator = pipeline("text-generation", model="takedakoji00//gs/bs/tga-aklab/ktakeda/tmp_dir_not_delete/tmpxegxll6g/takedakoji00/Llama-3.1-8B-Instruct-custom-qg-full_20250219-7th_random_pad_is_eos_offline_nav")
4outputs = generator("Hello, my llama is cute")1from transformers import AutoTokenizer
2from trl import AutoModelForCausalLMWithValueHead
3
4tokenizer = AutoTokenizer.from_pretrained("takedakoji00//gs/bs/tga-aklab/ktakeda/tmp_dir_not_delete/tmpxegxll6g/takedakoji00/Llama-3.1-8B-Instruct-custom-qg-full_20250219-7th_random_pad_is_eos_offline_nav")
5model = AutoModelForCausalLMWithValueHead.from_pretrained("takedakoji00//gs/bs/tga-aklab/ktakeda/tmp_dir_not_delete/tmpxegxll6g/takedakoji00/Llama-3.1-8B-Instruct-custom-qg-full_20250219-7th_random_pad_is_eos_offline_nav")
6
7inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
8outputs = model(**inputs, labels=inputs["input_ids"])