diff --git a/.dockerignore b/.dockerignore index 47d5479..6c06cc1 100644 --- a/.dockerignore +++ b/.dockerignore @@ -69,6 +69,16 @@ logs/ models/finetune_cv/ models/benchmark/ models/mpnn/ +models/qwen2.5-7b-instruct/ +models/cv5_sample/ +models/molt5-base/ +models/nested_cv/ +models/pretrain/ +MolE_ckpt/ +MolE_repo/ +models/abl/ +models/biot5-plus-base/ +models/abl_full/ models/*.pt models/*.json !models/final/ diff --git a/requirements.txt b/requirements.txt index 9bce2ae..0eb08c4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,3 +23,5 @@ captum>=0.7.0 transformers>=4.30,<4.46 sentencepiece protobuf +bitsandbytes +accelerate