* compute fid score code * cleaner s1 and m1 loading
* added model impl * minor cleanups * extracted weights loading into from_pretrained * reorganized model for better weight loading * removed lru cache for state dict loading