Add a task to import weights
Version bump