зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-30 13:36:04 +02:00
fixed blip2 model in test_multimodal_search.py
Этот коммит содержится в:
родитель
d7133e8dc3
Коммит
37c6d1ad77
@ -1,3 +1,4 @@
|
|||||||
|
import pytest
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
import numpy
|
import numpy
|
||||||
from torch import device, cuda, no_grad
|
from torch import device, cuda, no_grad
|
||||||
@ -23,7 +24,7 @@ def test_read_img():
|
|||||||
test_img = ms.MultimodalSearch.read_img(my_dict, TEST_IMAGE_2)
|
test_img = ms.MultimodalSearch.read_img(my_dict, TEST_IMAGE_2)
|
||||||
assert list(numpy.array(test_img)[257][34]) == [70, 66, 63]
|
assert list(numpy.array(test_img)[257][34]) == [70, 66, 63]
|
||||||
|
|
||||||
|
@pytest.mark.skipif(not cuda.is_available(), reason="model for gpu only")
|
||||||
def test_load_feature_extractor_model_blip2():
|
def test_load_feature_extractor_model_blip2():
|
||||||
my_dict = {}
|
my_dict = {}
|
||||||
multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
||||||
|
|||||||
Загрузка…
x
Ссылка в новой задаче
Block a user