fix: inference for Chroma model

2026-04-08 06:28:48 +00:00 · 2025-07-20 14:08:54 +09:00
parent 24d2ea86c7
commit 404ddb060d
3 changed files with 23 additions and 18 deletions
--- a/library/flux_utils.py
+++ b/library/flux_utils.py
@@ -146,7 +146,7 @@ def load_flow_model(
        from . import chroma_models

        # build model
-        logger.info("Building Chroma model from BFL checkpoint")
+        logger.info("Building Chroma model")
        with torch.device("meta"):
            model = chroma_models.Chroma(chroma_models.chroma_params)
            if dtype is not None: