Skip to content

Commit c7c0d13

Browse files
committed
fix test_modelscope.py
Signed-off-by: ZX-ModelCloud <zx@modelcloud.ai>
1 parent 28bbf69 commit c7c0d13

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/test_modelscope.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def setUpClass(self):
2121
def test_load_modelscope(self):
2222
model = GPTQModel.load(self.MODEL_ID)
2323

24-
result = model.generate("The capital of France is")[0]
24+
result = model.generate("The capital city of France is named")[0]
2525
str_output = model.tokenizer.decode(result)
2626
assert "paris" in str_output.lower() or "city" in str_output.lower()
2727

0 commit comments

Comments
 (0)