Quellcode durchsuchen

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

Nazrul Islam vor 5 Monaten
Ursprung
Commit
80db25e726
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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)