Downloads · 30 days
18
5% of all-time downloads
towardsinnovationlab/qwen3-medical
qwen3-medical is a text generation model from towardsinnovationlab. Use it when you need the model to write or continue text. It is set up for peft. The card lists the license as apache-2.0.
This is a LoRA adapter for Qwen/Qwen3-0.6B, trained using the Unsloth library for parameter-efficient fine-tuning (PEFT). When loaded via the transformers pipeline or PeftModel, the base model Qwen/Qwen3-0.6B is autom…
Downloads · 30 days
18
5% of all-time downloads
All-time downloads
384
Public
Repo size
173 MB
Likes
0
Public
Click a slice to open those files.
.safetensors80.8 MB · 84%
From the Hugging Face model README
This is a LoRA adapter for Qwen/Qwen3-0.6B, trained using the Unsloth library for parameter-efficient fine-tuning (PEFT). When loaded via the transformers pipeline or PeftModel, the base model Qwen/Qwen3-0.6B is automatically fetched, and the adapter is applied on top.
The adapter was trained on the following dataset:
-OpenMed/Medical-Reasoning-SFT-GPT-OSS-120B - for enhancing medical reasoning skills
This adapter can be used as support in healthcare applications, medical research, and clinical text generation.
It can be integrated into educational chatbots for medical reasoning conversations.
It is not suitable for high-level decision-making.
Conversational quality may degrade with complex or multi-turn inputs. The adapter may give inaccurate answers and should be referred to a professional.
Use the code below to get started with the adapter.
-Using the pipeline:
# Use a pipeline as a high-level helper
from transformers import pipeline
import re
pipe = pipeline("text-generation", model="towardsinnovationlab/qwen3-medical")
messages = [
{"role": "user", "content": "What are the main symptoms of heart disease? Please provide your answer in bullet points."},
]
result = pipe(messages)
# Extract only the assistant's response
assistant_response = result[0]['generated_text'][-1]['content']
# Remove the <think> tags and their content
clean_response = re.sub(r'<think>.*?</think>', '', assistant_response, flags=re.DOTALL).strip()
print(clean_response)
-Loading the model:
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
import re
tokenizer = AutoTokenizer.from_pretrained("towardsinnovationlab/qwen3-medical")
model = AutoModelForCausalLM.from_pretrained("towardsinnovationlab/qwen3-medical")
messages = [
{"role": "user", "content": "What are the main symptoms of heart disease? Please provide your answer in bullet points."},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(
**inputs,
max_new_tokens=512,
temperature=0.7,
top_p=0.8,
top_k=20,
do_sample=True
)
# Extract assistant's response
assistant_response = tokenizer.decode(
outputs[0][inputs["input_ids"].shape[-1]:],
skip_special_tokens=True
)
# Remove <think> tags and their content
clean_response = re.sub(r'<think>.*?</think>', '', assistant_response, flags=re.DOTALL).strip()
print(clean_response)
-OpenMed/Medical-Reasoning-SFT-GPT-OSS-120B with 200,193 synthetic medical conversations.
-Colab with NVIDIA A100 GPU
-per_device_train_batch_size = 4,
-gradient_accumulation_steps = 8,
-warmup_steps = 5,
-max_steps = 30,
-learning_rate = 2e-5,
-logging_steps = 100,
-save_steps=500,
-optim = "adamw_8bit",
-weight_decay = 0.001,
-lr_scheduler_type = "linear"
Loss Value >> 1.4693