зеркало из
				https://github.com/ssciwr/AMMICO.git
				synced 2025-10-30 21:46: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
	 Petr Andriushchenko
						Petr Andriushchenko