Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion lightx2v/models/runners/cosmos3/cosmos3_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ def load_text_encoder(self):
if not os.path.exists(tokenizer_path):
tokenizer_path = self.config["model_path"]
logger.info(f"Loading Cosmos3 tokenizer from {tokenizer_path}")
return AutoTokenizer.from_pretrained(tokenizer_path, trust_remote_code=True)
return AutoTokenizer.from_pretrained(tokenizer_path, trust_remote_code=self.config.get("trust_remote_code", False))

def load_image_encoder(self):
return None
Expand Down
27 changes: 27 additions & 0 deletions tests/test_cosmos3_trust_remote_code.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
from __future__ import annotations

from unittest.mock import patch

from lightx2v.models.runners.cosmos3.cosmos3_runner import Cosmos3Runner


def _runner_with_config(config: dict) -> Cosmos3Runner:
runner = Cosmos3Runner.__new__(Cosmos3Runner)
runner.config = config
return runner


def test_load_text_encoder_defaults_trust_remote_code_to_false(tmp_path):
config = {"model_path": str(tmp_path)}
runner = _runner_with_config(config)
with patch("lightx2v.models.runners.cosmos3.cosmos3_runner.AutoTokenizer.from_pretrained") as from_pretrained:
runner.load_text_encoder()
assert from_pretrained.call_args.kwargs["trust_remote_code"] is False


def test_load_text_encoder_honors_explicit_trust_remote_code_opt_in(tmp_path):
config = {"model_path": str(tmp_path), "trust_remote_code": True}
runner = _runner_with_config(config)
with patch("lightx2v.models.runners.cosmos3.cosmos3_runner.AutoTokenizer.from_pretrained") as from_pretrained:
runner.load_text_encoder()
assert from_pretrained.call_args.kwargs["trust_remote_code"] is True