File size: 1,124 Bytes
30d9278 0a9e0c4 30d9278 283238a 30d9278 75a22b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# Bio-SIEVE
[Arxiv](https://arxiv.org/abs/2308.06610) [Github](https://github.com/ambroser53/Bio-SIEVE)
The single-task Bio-SIEVE optimised for Inclusion/Exclusion classification of biomedical literature given the systematic review's objectives and selection criteria.
These are LoRA weights, having continued training from Guanaco7B. The base LLaMA model will be automatically downloaded when used with the generate_cli.py script from our [Github](https://github.com/ambroser53/Bio-SIEVE).
Also see the [multi-task variant](https://huggingface.co/Ambroser53/Bio-SIEVE-Multi) for exclusion reasoning and PIO extraction. Variants trained without instruction tuned pretraining TBD.
---
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: float32
### Framework versions
- PEFT 0.4.0.dev0
|