tt-model push embeddinggemma-2-p150 (container)

714c70a verified 3 days ago

421 Bytes

# SPDX-FileCopyrightText: © 2026 Tenstorrent USA, Inc.

# SPDX-License-Identifier: Apache-2.0

from models.autoports.google_embeddinggemma_2.tt.fused_encoder_layer import FusedEncoderLayer


class FusedDecoder(FusedEncoderLayer):
    def prefill_forward(self, hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window=None):
        return self.forward(hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window)

	# SPDX-FileCopyrightText: © 2026 Tenstorrent USA, Inc.

	# SPDX-License-Identifier: Apache-2.0

	from models.autoports.google_embeddinggemma_2.tt.fused_encoder_layer import FusedEncoderLayer


	class FusedDecoder(FusedEncoderLayer):
	def prefill_forward(self, hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window=None):
	return self.forward(hidden, per_layer_input, attn_mask, cos, sin, plan, cu_window)