Is this project tensorrt compatible #337
Description
has any body tired to use this x_transformer in an AMP training project? is it the function def apply_rotary_pos_emb(t, freqs, scale = 1): compatible with Tensorrt fp16 format model conversion?
{{ message }}