Function inject_model

tricks/modules/ltx_model.py:250–257 · view source on GitHub ↗

(diffusion_model)

Source from the content-addressed store, hash-verified

248
249
250	def inject_model(diffusion_model):
251	diffusion_model.__class__ = LTXVModelModified
252	for idx, transformer_block in enumerate(diffusion_model.transformer_blocks):
253	transformer_block.__class__ = LTXModifiedBasicTransformerBlock
254	transformer_block.idx = idx
255	transformer_block.attn1.__class__ = LTXModifiedCrossAttention
256	transformer_block.attn1.idx = idx
257	return diffusion_model

modifyMethod · 0.85

no outgoing calls

no test coverage detected