Update modeling_tinyllava_phi.py

modeling_tinyllava_phi.py

@@ -1,12 +1,22 @@
 # For licensing see accompanying LICENSE file.
-# Copyright (C) 2024 Apple Inc. All Rights Reserved.
+# Copyright (C) 2024 TinyLLaVA. All Rights Reserved.
+import time
 
+import dataclasses
+from enum import auto, Enum
+from typing import List, Tuple, Optional, Union
+import requests
+from PIL import Image
+from io import BytesIO
+import base64
 import re
+
 import torch
 import torch.utils.checkpoint
 from torch import nn
 from torch.nn import functional as F
 
+from transformers.utils import logging
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutputWithPast
 from transformers.generation.utils import GenerateOutput
@@ -16,19 +26,7 @@ from .configuration import TinyLlavaConfig, IGNORE_INDEX, IMAGE_TOKEN_INDEX, DEF
 
 from transformers import AutoConfig, AutoModelForCausalLM, PhiForCausalLM
 
-import time
-from transformers.utils import logging
-import dataclasses
-from enum import auto, Enum
-from typing import List, Tuple
 
-# this import has to be relative, otherwise, when setting trust_remote_code=True
-# huggingface transformers won't be able to load the module correctly
-from typing import List, Optional, Union
-import requests
-from PIL import Image
-from io import BytesIO
-import base64
 
 logger = logging.get_logger(__name__)