error on inference using DNABERT2 - can you please share the environment you used for running it?

#12
by NettaB - opened

Hi,
Can you please share the env or the versions of the main packages used for running DNABERT2? (python version, pytorch, transformers, triton etc)
I saw in github that you recommended installing triton from source code but in one of the issues you recommended installing a specific version using pip.
Thanks!

I'm not sure if what made you write this is the problem about the "config", if it's the case, then use
config = BertConfig.from_pretrained("zhihan1996/DNABERT-2-117M")
model = AutoModel.from_config(config)
instead of
model = AutoModel.from_pretrained("zhihan1996/DNABERT-2-117M", trust_remote_code=True)
and it should work.
I just run the first part of the README and it works well having the latest versions of each requirement except for transformers==4.29.2

Sign up or log in to comment