Problem
It is currently impossible to create the model via madeleine.models.factory.create_model_from_pretrained without cuda availability due to the following line.
madeleine.models.factory import create_model_from_pretrained
Fix
If I don't miss anything, this should be fixable by simply adding an argument to the create_model_from_pretrained.
Keep up the fantastic work at Mahmood Lab!
Problem
It is currently impossible to create the model via
madeleine.models.factory.create_model_from_pretrainedwithout cuda availability due to the following line.madeleine.models.factory import create_model_from_pretrained
Fix
If I don't miss anything, this should be fixable by simply adding an argument to the
create_model_from_pretrained.Keep up the fantastic work at Mahmood Lab!