Views
No views yet
British Breakfast with baked beans, fried eggs, black pudding, sausages, bacon, mushrooms, a cup of tea and toast and fried tomatoesfood_or_drink: 1
tags: fi, di
foods: British Breakfast, baked beans, fried eggs, black pudding, sausages, bacon, mushrooms, toast, fried tomatoes
drinks: teagpt-oss-120b.| Tag | Meaning |
|---|---|
| np | nutrition_panel |
| il | ingredient_list |
| me | menu |
| re | recipe |
| fi | food_items |
| di | drink_items |
| fa | food_advertisement |
| fp | food_packaging |
transformers + trl (SFTTrainer)1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2
3MODEL_PATH = "OmarShamakh/FoodExtract-gemma-3-270m-fine-tune-v1"
4
5model = AutoModelForCausalLM.from_pretrained(
6 MODEL_PATH, dtype="auto", device_map="auto", attn_implementation="eager"
7)
8tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
9
10pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
11
12input_text = "A plate with bacon, eggs and toast on it"
13input_prompt = pipe.tokenizer.apply_chat_template(
14 conversation=[{"role": "user", "content": input_text}],
15 tokenize=False,
16 add_generation_prompt=True
17)
18output = pipe(input_prompt, max_new_tokens=256)
19print(output[0]["generated_text"][len(input_prompt):])