xdecoder/Instruct-X-Decoder
163
1U
2 E$�c � @ sX d dl mZ d dlZd dlmZ ddlmZ ddlmZ G dd� de�Z edd � �Z3dS )4� )�requiresN� )�register_model)�LanguageEncoderc sl e Zd Z� fdd�Ze�� � fdd��Ze�� � fdd��Ze�� � fdd��Ze�� � fd d5��Z � Z6S ) �FixLanguageEncoderc s, t t| �j||� tjt�g �dd�| _d S )NF)�
requires_grad)�superr �__init__�nn� Parameter�torch�ones� logit_scale��self�args�kwargs�� __class__� �8/data/arXiv/demo/Demo/xdecoder/language/fixvlpencoder.pyr 7 s zFixLanguageEncoder.__init__c s t � j||�S �N)r �get_text_embeddingsr r r r r s z&FixLanguageEncoder.get_text_embeddingsc s t � j||�S r )r �get_text_token_embeddingsr r r r r s z,FixLanguageEncoder.get_text_token_embeddingsc s t � j||�S r )r �forward_languager r r r r s z#FixLanguageEncoder.forward_languagec s t � j||�S r )r �forward_language_tokenr r r r r s z)FixLanguageEncoder.forward_language_token) �__name__�8__module__� __qualname__r r �no_gradr r r r �
__classcell__r r r r r s r c K s t | �S r )r )�cfgr r r r �get_language_model! s r" ) �importlib.metadatar r �torch.nnr9 �registryr �10vlpencoderr r r" r r r r �<module> s