fix: use model name model.safetensors

inference.py

@@ -18,7 +18,6 @@ from litgpt.generate.base import (
 )
 import soundfile as sf
 from litgpt.model import GPT, Config
-from lightning.fabric.utilities.load import _lazy_load as lazy_load
 from utils.snac_utils import layershift, reconscruct_snac, reconstruct_tensors, get_time_str
 from utils.snac_utils import get_snac, generate_audio_data
 import whisper
@@ -359,7 +358,7 @@ def load_model(ckpt_dir, device):
         model = GPT(config)
 
     model = fabric.setup(model)
-    state_dict = load_file(ckpt_dir + "/lit_model.safetensors")
+    state_dict = load_file(ckpt_dir + "/model.safetensors")
     model.load_state_dict(state_dict, strict=True)
     model.to(device).eval()