MCPcopy
hub / github.com/Yuanshi9815/OminiControl / _feedforward_forward

Function _feedforward_forward

omini/pipeline/flux_omini.py:203–211  ·  view source on GitHub ↗

FeedForward.forward routing each sub-module through ``lora_forward``. Only the output projection (``ff.net[2]``) is LoRA-wrapped in practice; ``lora_forward`` transparently falls back to a plain call for the others.

(ff, x, adapter)

Source from the content-addressed store, hash-verified

201
202
203def _feedforward_forward(ff, x, adapter):
204 """FeedForward.forward routing each sub-module through ``lora_forward``.
205
206 Only the output projection (``ff.net[2]``) is LoRA-wrapped in practice;
207 ``lora_forward`` transparently falls back to a plain call for the others.
208 """
209 for module in ff.net:
210 x = lora_forward(module, x, adapter)
211 return x
212
213
214def attn_forward(

Callers 1

block_forwardFunction · 0.85

Calls 1

lora_forwardFunction · 0.85

Tested by

no test coverage detected