license: llama2
language:
- en
pipeline_tag: text-generation
tags:
- endocrinology
- medical
- llama
- thyroid
- diabetes
- pituitary
- adrenal
library_name: peft
widget:
- text: |
### Question: What is hypothyroidism?
### Input:
### Response:
inference:
parameters:
temperature: 0.1
max_new_tokens: 1024
Document created on September 7 2023
Endo_Llama: Thyroid Edition (v1)
Model Description
What is it?
Endo_Llama is a large language model (LLama 2) finetuned to answer questions related to endocrinology, with a focus on thyroid-related issues in its first version. This model aims to assist healthcare professionals and researchers in obtaining better information about thyroid conditions, treatments, and best practices.
Input Models input text only.
Output Models generate text only.
Model Architecture
Model Type: Llama 2 Parameters: ~7B Training Data: Curated dataset of endocrinology questions and answers, with a focus on thyroid issues for Version 1.
Intended Use
Clinical decision support Medical research
Limitations
Not a substitute for professional medical advice May not cover all nuances of a complex medical condition Version 1 is primarily focused on thyroid-related issues
Training Data
The model is trained on a curated dataset of endocrinology questions and answers, with an emphasis on thyroid conditions.
Usage
coming soon!
library_name: peft
Training procedure
The following bitsandbytes quantization config was used during training:
- load_in_8bit: False
- load_in_4bit: True
- llm_int8_threshold: 6.0
- llm_int8_skip_modules: None
- llm_int8_enable_fp32_cpu_offload: False
- llm_int8_has_fp16_weight: False
- bnb_4bit_quant_type: nf4
- bnb_4bit_use_double_quant: True
- bnb_4bit_compute_dtype: float16
Framework versions
PEFT 0.5.0
PEFT 0.5.0