Hugging Face Github missmatch

[JK-rez]

model_loader.py in Hugging Face and GitHub are different, with github version containing code that will run when importing the class.
There is also no import torch.nn as nn in either version, making net.classifier[2] = nn.Identity() fail.

[morgengramlich]

Sorry, I kinda abandoned this repo and wasn't doing anything ML-related for quite some time. What's the issue exactly? I don't think I use nn.Identity() here