o }o™i©ã@s<ddlZddlZddlmZddlmZdd„Zdd„ZdS)éN)Úvlm)ÚDownSampleBlockcCs,tjdd}|jdks J‚|jdksJ‚dS)NÚsiglip)Úvision_model_typeFr)rÚ CLIPViTConfigÚadd_class_tokenÚclass_token_len)Úconfig©r úZ/home/ubuntu/.local/lib/python3.10/site-packages/tests/collections/vlm/vision/test_base.pyÚtest_siglip_config_errors r cCsFd}d}d}t |||¡}tƒ}||ƒ}td|jƒ|jdks!J‚dS)Néééz Output shape:)réé@)ÚtorchÚrandnrÚprintÚshape)Ú batch_sizeÚseq_lenÚ embed_dimÚxÚblockÚoutr r r Útest_downsample_block_basics r)ÚpytestrÚnemo.collectionsrÚ nemo.collections.vlm.vision.baserr rr r r r Ús