Selaa lähdekoodia

fix: clarify model weight loading log message (#1127)

Nazrul Islam 5 kuukautta sitten
vanhempi
commit
80db25e726
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      fish_speech/models/text2semantic/llama.py

+ 1 - 1
fish_speech/models/text2semantic/llama.py

@@ -491,7 +491,7 @@ class BaseTransformer(nn.Module):
                     )
 
             err = model.load_state_dict(weights, strict=False, assign=True)
-            logger.info(f"Loaded weights with error: {err}")
+            logger.info(f"Model weights loaded - Status: {err}")
 
         if lora_config is not None:
             setup_lora(model, lora_config)