change header

CASIA-IVA-Lab · Aug 29, 2023 · 14bf5d0 · 14bf5d0
1 parent 2558043
commit 14bf5d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/header.py b/code/header.py
@@ -25,7 +25,7 @@
 from copy import deepcopy
 import ipdb
 import argparse
-import data
+from model.ImageBind import data
 from transformers import LlamaTokenizer, LlamaForCausalLM, LlamaConfig
 from torch.nn.utils.rnn import pad_sequence
 from peft import LoraConfig, TaskType, get_peft_model