dsgiitr / AIKavach

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Make the model loading method universal

mr-sarthakgupta opened this issue · comments

Currently, the model input and loading method breaks whenever the loaded model requires some different dependencies or needs to be loaded from model class and weight dicts. We need to make a general system to help input and load the models.