mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-23 09:09:01 +03:00
skip these case for no-cuda device since the modeling contains cuda s… (#14656)
skip these case for no-cuda device since the modeling contains cuda specific code see https://huggingface.co/krea/krea-realtime-video/blob/main/transformer/model.py#L33 Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
This commit is contained in:
@@ -36,7 +36,7 @@ from diffusers.modular_pipelines import (
|
||||
)
|
||||
from diffusers.utils import logging
|
||||
|
||||
from ..testing_utils import CaptureLogger, nightly, require_torch, require_torch_accelerator, slow, torch_device
|
||||
from ..testing_utils import CaptureLogger, nightly, require_torch_accelerator, require_torch_gpu, slow, torch_device
|
||||
|
||||
|
||||
def _create_tiny_model_dir(model_dir):
|
||||
@@ -722,7 +722,7 @@ class TestBlockKwargsTypeInputs:
|
||||
|
||||
@slow
|
||||
@nightly
|
||||
@require_torch
|
||||
@require_torch_gpu
|
||||
class TestKreaCustomBlocksIntegration:
|
||||
repo_id = "krea/krea-realtime-video"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user