Tuned model loading, confirming reproducibility on HF

Second commit, confirming reproducibility on HF

First commit, confirming reproducibility on HF

initial commit