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

714c70a verified 4 days ago

437 Bytes

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

# SPDX-License-Identifier: Apache-2.0

from models.autoports.google_embeddinggemma_2.tt.optimized_encoder_layer import OptimizedEncoderLayer


class OptimizedDecoder(OptimizedEncoderLayer):
    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.optimized_encoder_layer import OptimizedEncoderLayer


	class OptimizedDecoder(OptimizedEncoderLayer):
	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)