Procházet zdrojové kódy

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

Nazrul Islam před 5 měsíci
rodič
revize
80db25e726
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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)