Views
No views yet
pip install transformers>=4.37.0 torchtransformers library:1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained("kulia-moon/Lily-Qwen1.5-0.5B", torch_dtype="auto", device_map="auto")
5tokenizer = AutoTokenizer.from_pretrained("kulia-moon/Lily-Qwen1.5-0.5B")1# Define prompt
2prompt = "Write a short story about a magical garden."
3
4# Create chat template
5messages = [
6 {"role": "system", "content": "You are a creative assistant with a passion for storytelling."},
7 {"role": "user", "content": prompt}
8]
9
10# Apply chat template
11text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
12
13# Tokenize input
14model_inputs = tokenizer([text], return_tensors="pt")
15
16# Generate response
17generated_ids = model.generate(model_inputs.input_ids, max_new_tokens=512)
18
19# Decode output
20response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
21print(response)In a hidden valley, there bloomed a magical garden where flowers sang softly under the moonlight. Lily, a young explorer, stumbled upon it one evening...
1# Define conversation
2messages = [
3 {"role": "system", "content": "You are Lily, a friendly assistant who loves nature."},
4 {"role": "user", "content": "What's your favorite flower?"}
5]
6
7# Apply chat template
8text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
9
10# Tokenize and generate
11model_inputs = tokenizer([text], return_tensors="pt")
12generated_ids = model.generate(model_inputs.input_ids, max_new_tokens=100)
13
14# Decode response
15response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
16print(response)Oh, I adore cherry blossoms! Their delicate pink petals remind me of spring's gentle embrace.
1# Define translation prompt
2prompt = "Translate 'The garden blooms with vibrant colors' into Chinese."
3
4messages = [
5 {"role": "system", "content": "You are a multilingual assistant."},
6 {"role": "user", "content": prompt}
7]
8
9# Apply chat template
10text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
11
12# Tokenize and generate
13model_inputs = tokenizer([text], return_tensors="pt")
14generated_ids = model.generate(model_inputs.input_ids, max_new_tokens=50)
15
16# Decode response
17response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
18print(response)花园盛开着鲜艳的色彩。
1from transformers import Trainer, TrainingArguments, DataCollatorForLanguageModeling
2from datasets import load_dataset
3
4# Load dataset
5dataset = load_dataset("your_dataset")
6
7# Define training arguments
8training_args = TrainingArguments(
9 output_dir="./lily-finetuned",
10 per_device_train_batch_size=4,
11 num_train_epochs=3,
12 learning_rate=5e-5,
13)
14
15# Initialize trainer
16trainer = Trainer(
17 model=model,
18 args=training_args,
19 train_dataset=dataset["train"],
20 data_collator=DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False),
21)
22
23# Start fine-tuning
24trainer.train()BETA1@misc{lily-qwen1.5-0.5b,
2 author = {kulia-moon},
3 title = {Lily-Qwen1.5-0.5B: A Fine-Tuned Qwen1.5 Model},
4 year = {2025},
5 publisher = {Hugging Face},
6 journal = {Hugging Face Model Hub}
7}