support OMP_NUM_THREADS for CPU limit

[alexweisberg]

Hi, I am looking into an issue with a separate program that uses tape (zhangyumeng1sjtu/T4SEfinder#2). I am running this program on a computing cluster that does not have a GPU, and so I need to limit the number of threads spawned by each program to a specific number. The program I am using is called T4SEfinder, which calls the transformer function of tape as follows:
tape-embed transformer {fasta} {embed_path} {pretrained_model} --tokenizer iupac --batch_size 1 --seed {random_seed} --log_level WARNING
and that is spawning as many processes as the machine has CPUs.

Would it be possible for tape to respect the OMP_NUM_THREADS environment variable when running? Or alternatively adding a number of threads argument? Thanks!