fix(inference): update condition for loading models (#221)

fix(inference): update condition for loading models (#221) #409