fix: 修复参数传递问题

2026-03-30 22:22:36 +08:00
parent e1f9901384
commit 345fd2f091
6 changed files with 13 additions and 5 deletions
@@ -96,7 +96,7 @@ def base_test_env(request: pytest.FixtureRequest):
    
    with open(config_path, 'w') as f:
        json.dump(config, f)
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
    transformer_config = ModelConfig().load(config_path)
    model = Transformer(transformer_config).to(device=device)
    tokenizer = BpeTokenizer()