зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-29 13:06:04 +02:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Этот коммит содержится в:
родитель
097142fd36
Коммит
a6ea571478
@ -22,27 +22,9 @@ def test_read_img():
|
||||
test_img = ms.read_img(TEST_IMAGE_2)
|
||||
assert numpy.array(test_img)[257][34] == [70, 66, 63]
|
||||
|
||||
|
||||
|
||||
#def test_load_feature_extractor_model_blip2():
|
||||
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
||||
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
# def test_load_feature_extractor_model_blip2():
|
||||
# multimodal_device = device("cuda" if cuda.is_available() else "cpu")
|
||||
# (model, vis_processors, txt_processors,) = ms.load_feature_extractor_model_blip2(multimodal_device)
|
||||
|
||||
mydict = {}
|
||||
for img_path in images:
|
||||
@ -54,4 +36,4 @@ def test_read_img():
|
||||
keys = list(mydict.keys())
|
||||
assert len(mydict) == 12
|
||||
for key in keys:
|
||||
assert len(mydict[key]["3_non-deterministic summary"]) == 3
|
||||
assert len(mydict[key]["3_non-deterministic summary"]) == 3
|
||||
|
||||
Загрузка…
x
Ссылка в новой задаче
Block a user