File tree 1 file changed +5
-1
lines changed
xinference/model/video/tests
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 13
13
# limitations under the License.
14
14
import logging
15
15
16
+ import pytest
17
+
16
18
from .. import BUILTIN_VIDEO_MODELS
17
19
from ..core import cache
18
20
from ..diffusers import DiffUsersVideoModel
19
21
20
22
logger = logging .getLogger (__name__ )
21
23
22
24
25
+ @pytest .mark .skip (reason = "Video model requires too many GRAM." )
23
26
def test_model ():
24
27
test_model_spec = next (iter (BUILTIN_VIDEO_MODELS .values ()))
25
28
model_path = cache (test_model_spec )
@@ -31,6 +34,7 @@ def test_model():
31
34
assert r
32
35
33
36
37
+ @pytest .mark .skip (reason = "Video model requires too many GRAM." )
34
38
def test_client (setup ):
35
39
endpoint , _ = setup
36
40
from ....client import Client
@@ -56,4 +60,4 @@ def test_client(setup):
56
60
"The background includes a small, flowing stream and vibrant green foliage, "
57
61
"enhancing the peaceful and magical atmosphere of this unique musical performance."
58
62
)
59
- print ( r )
63
+ assert r
You can’t perform that action at this time.
0 commit comments