Method _get_layer

diffsynth/models/kolors_text_encoder.py:863–864 · view source on GitHub ↗

(self, layer_number)

Source from the content-addressed store, hash-verified

861	self.gradient_checkpointing = False
862
863	def _get_layer(self, layer_number):
864	return self.layers[layer_number]
865
866	def forward(
867	self, hidden_states, attention_mask, rotary_pos_emb, kv_caches=None,

forwardMethod · 0.95

no outgoing calls

no test coverage detected