Bio-SIEVE
The single-task Bio-SIEVE optimised for Inclusion/Exclusion classification of biomedical literature given the systematic review's objectives and selection criteria.
There are LoRA weights, having continued training from Guanaco. The base LLaMA model will be automatically downloaded when used with the generate_cli.py script from our Github.
Also see the multi-task variant 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