bug : cannot trace HDiT in export scripts

[fchouteau]

HDiT arch is not traceable,

Fixes :

1. K_DIFFUSION_USE_COMPILE should be set to 0 at export time in order to bypass the torch.compile ops that can't be traced

models/modules/hdit/flags.py#L9

2. apply_rotary_embed_ is not traceable models/modules/hdit/hdit.py#L218

However apply_rotary_embed models/modules/hdit/hdit.py#L202

So it should be a matter of monkey patching the two functions at export time :

Example :

def export(opt, cuda, model_in_file, model_out_file, opset_version, export_type):

    model = gan_networks.define_G(**vars(opt))

    if isinstance(model, models.modules.hdit.hdit.HDiT):
        models.modules.hdit.hdit.apply_rotary_emb_ = (
            models.modules.hdit.hdit.apply_rotary_emb
        )

PR should be up in the following days