🐛 Correctly unset loaded state after unloading

✨ Implement lazy loading for models and correct tokens counting