Datasetpublic

ysn-rfd/text-dataset-tiny-code-script-py-format

USED of tahamajs/medicine_ds_persian for .parquet file USED of Alijafarixcs2/persian-it-llama2-2k for .parquet file USED of Abirate/english_quotes for .jsonl file NEW FILES (05/12/2025) NEW FILES (12/26/2025) NEW FILES (02/15/2026)

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face

3likes1.6kdownloads

rootpytorch_fine_tuning_codefine_tuning_tiny_t5.py

fine_tuning_tiny_t5.py20 linesDownload Raw Back to pytorch_fine_tuning_code

1import torch
2from transformers import pipeline
3
4model_id = "prithivMLmods/Bellatrix-Tiny-0.5B"
5pipe = pipeline(
6    "text-generation",
7    model=model_id,
8    torch_dtype=torch.float16,
9    device_map="cpu",
10)
11messages = [
12    {"role": "system", "content": "You are a think chatbot"},
13    {"role": "user", "content": "hp elite book 840 g3"},
14]
15outputs = pipe(
16    messages,
17    max_new_tokens=16,
18)
19print(outputs[0]["generated_text"][-1])
20