Skip to content

Commit f5cbbd6

Browse files
codingl2k1qinxuye
authored andcommitted
Fix
1 parent 2308d23 commit f5cbbd6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

xinference/model/video/tests/test_diffusers_video.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,16 @@
1313
# limitations under the License.
1414
import logging
1515

16+
import pytest
17+
1618
from .. import BUILTIN_VIDEO_MODELS
1719
from ..core import cache
1820
from ..diffusers import DiffUsersVideoModel
1921

2022
logger = logging.getLogger(__name__)
2123

2224

25+
@pytest.mark.skip(reason="Video model requires too many GRAM.")
2326
def test_model():
2427
test_model_spec = next(iter(BUILTIN_VIDEO_MODELS.values()))
2528
model_path = cache(test_model_spec)
@@ -31,6 +34,7 @@ def test_model():
3134
assert r
3235

3336

37+
@pytest.mark.skip(reason="Video model requires too many GRAM.")
3438
def test_client(setup):
3539
endpoint, _ = setup
3640
from ....client import Client
@@ -56,4 +60,4 @@ def test_client(setup):
5660
"The background includes a small, flowing stream and vibrant green foliage, "
5761
"enhancing the peaceful and magical atmosphere of this unique musical performance."
5862
)
59-
print(r)
63+
assert r

0 commit comments

Comments
 (0)