How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="NewstaR/StableGalen-6b", trust_remote_code=True)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("NewstaR/StableGalen-6b", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("NewstaR/StableGalen-6b", trust_remote_code=True)
Quick Links

This model is the DeciLM-6b-Instruct model, trained specifically for medicine

Galen uses the

### User: {prompt}

### Response:

or

{prompt} 

Prompt templates

Galen Training Recipe:

  • target_modules = ["q_proj", "v_proj", "gate_proj", "down_proj", "up_proj", "k_proj", "o_proj"]
  • Learning Rate: 4e-4
  • LR Scheduler: constant
  • 250 StepsLoss

T3: 1 Hour

Downloads last month
932
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train NewstaR/StableGalen-6b