Downloads · 30 days
0
fine-tune-deep-seek-r1
Vedant101/fine-tune-deep-seek-r1
fine-tune-deep-seek-r1 is a machine learning model from Vedant101. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This model is a fine-tuned version of DeepSeek R1 specialized for medical reasoning and clinical decision-making.
Access
Public
Updated Feb 2, 2025
Repo size
185 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors168 MB · 91%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Feb 1, 2025
- Updated
- Feb 2, 2025
- SHA
- 9f698c58
Model card
From the Hugging Face model README
Fine-tuned DeepSeek R1 Model for Medical Reasoning
This model is a fine-tuned version of DeepSeek R1 specialized for medical reasoning and clinical decision-making.
Training Details
- Base Model: unsloth/DeepSeek-R1-Distill-Llama-8B
- Training Data: Medical reasoning dataset (FreedomIntelligence/medical-o1-reasoning-SFT)
- Fine-tuning Method: LoRA (Low-Rank Adaptation)
- Training Parameters:
- Batch Size: 2
- Learning Rate: 2e-4
- Epochs: 1
- Max Sequence Length: 2048
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("Vedant101/fine-tune-deep-seek-r1")
tokenizer = AutoTokenizer.from_pretrained("Vedant101/fine-tune-deep-seek-r1")
- License
- apache-2.0
- Languages
- en
- Created
- Feb 1, 2025
- Updated
- Feb 2, 2025