Update model/openlamm.py

model/openlamm.py

@@ -11,7 +11,7 @@ from torch.nn.utils import rnn
 from types import SimpleNamespace
 from peft import LoraConfig, TaskType, get_peft_model
 from transformers import LlamaTokenizer, LlamaForCausalLM, LlamaConfig
-import conversations
+
 import numpy as np
 # from header import *
 
@@ -19,9 +19,9 @@ from transformers import StoppingCriteria, StoppingCriteriaList
 
 from .CLIP import load as load_clip
 from .PROCESS import data
-
 from .modeling_llama import LlamaForCausalLM
 from .utils.pcl_utils import MEAN_COLOR_RGB, RandomCuboid, random_sampling
+import .conversations
 
 ImageFile.LOAD_TRUNCATED_IMAGES = True