Method get_embedding

tests/test_adapters.py:116–116

(self, model)

            def get_lm_head(self, model): return nn.Linear(1, 1)
            def get_router_input_dim(self, model): return 1
            def get_embedding(self, model): return nn.Embedding(1, 1)

        register_adapter("CustomModel", CustomAdapter)
        try: