Fix llava format (#751)

## Summary
<!--- This is a required section; please describe the main purpose of
this proposed code change. --->

<!---
## Details
This is an optional section; is there anything specific that reviewers
should be aware of?
--->

## Testing Done
<!--- This is a required section; please describe how this change was
tested. --->

<!-- 
Replace BLANK with your device type. For example, A100-80G-PCIe

Complete the following tasks before sending your PR, and replace `[ ]`
with
`[x]` to indicate you have done them. 
-->

- Hardware Type: <BLANK>
- [ ] run `make test` to ensure correctness
- [ ] run `make checkstyle` to ensure code style
- [ ] run `make test-convergence` to ensure convergence

Signed-off-by: Tcc0403 <[email protected]>

Author: Tcc0403

src/liger_kernel/transformers/model/llava.py

@@ -7,6 +7,7 @@
 
 from torch.nn import CrossEntropyLoss
 from transformers.models.llava.modeling_llava import LlavaCausalLMOutputWithPast
+from transformers.utils import is_torchdynamo_compiling
 
 from liger_kernel.transformers.fused_linear_cross_entropy import LigerFusedLinearCrossEntropyLoss
 from liger_kernel.transformers.model.loss_utils import LigerForCausalLMLoss
@@ -193,6 +194,7 @@ def lce_forward_deprecated(
         image_hidden_states=image_features if pixel_values is not None else None,
     )
 
+
 def lce_forward(
     self,
     input_ids: torch.LongTensor = None,
@@ -316,7 +318,6 @@ def lce_forward(
                 logits=logits, labels=labels, vocab_size=self.config.text_config.vocab_size, **lm_kwargs
             )
 
-
     if not return_dict:
         output = (logits,) + outputs[1:]
         return (loss,) + output if loss is not None else output

src/liger_kernel/transformers/monkey_patch.py

@@ -316,7 +316,7 @@ def apply_liger_kernel_to_llava(
     if fused_linear_cross_entropy:
         if transformer_version >= version.parse("4.52.0"):
             modeling_llava.LlavaForConditionalGeneration.forward = llava_lce_forward
-        elif transformer_version >= version.parse("4.49.0") and transformer_version < version.parse("4.52.0"): 
+        elif transformer_version >= version.parse("4.49.0") and transformer_version < version.parse("4.52.0"):
             modeling_llava.LlavaForConditionalGeneration.forward = llava_lce_forward_deprecated
         else:  # if version < 4.49.0
             logger.warning(