You can deploy this model and start earning money today!

venoda/
qwen3-0.6b-lora

Name: qwen3-0.6b-lora
Availability: InStock

tensorboard

safetensors

Views

No views yet

概要

「Mostly Basic Python Problems Dataset」を使用してLoRAを作成してみました。

使用方法

python

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import LoraConfig, TaskType, get_peft_model, PeftModel

device = torch.accelerator.current_accelerator().type if hasattr(torch, "accelerator") else "cuda"

model_id = "Qwen/Qwen3-0.6B"
tokenizer = AutoTokenizer.from_pretrained(model_id)

model = AutoModelForCausalLM.from_pretrained(model_id, device_map=device)
model = PeftModel.from_pretrained(model, "qwen3-0.6b-lora")

inputs = tokenizer("Preheat the oven to 350 degrees and place the cookie dough", return_tensors="pt")
outputs = model.generate(**inputs.to(device), max_new_tokens=50)

print(tokenizer.decode(outputs[0], skip_special_tokens=True))

venoda/qwen3-0.6b-lora

概要

使用方法

venoda/
qwen3-0.6b-lora