Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
codingl2k1 committed Aug 8, 2024
1 parent 6180cb6 commit b3a59d3
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion xinference/model/video/tests/test_diffusers_video.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,16 @@
# limitations under the License.
import logging

import pytest

from .. import BUILTIN_VIDEO_MODELS
from ..core import cache
from ..diffusers import DiffUsersVideoModel

logger = logging.getLogger(__name__)


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


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

0 comments on commit b3a59d3

Please sign in to comment.