Question: How to add the Dense layer to the model?
#2
by
thies
- opened
What I'm doing:
AutoModel.from_pretrained('sentence-transformers/distiluse-base-multilingual-cased-v1')
But this doesn't include the weights of the Dense layer.
I also tried:AutoModel.from_pretrained('sentence-transformers/distiluse-base-multilingual-cased-v1/2_Dense')
to load the Dense layer, but this doesn't work.
Is there a way to load those weights (apart from manually)?
Maybe you can use the interface provided by torch to load the dence model
It is recommended to use Sentence Transformers directly: https://huggingface.co/sentence-transformers/distiluse-base-multilingual-cased-v1#usage-sentence-transformers
tomaarsen
changed discussion status to
closed