Upload 18 files

inference.py

@@ -43,7 +43,7 @@ def main(args):
         stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)
         streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, timeout=20.0)
         # 加载图像
-        image = Image.open('./data/chang_chen.jpg').convert("RGB")
+        image = Image.open('./chang_chen.jpg').convert("RGB")
         image_tensor = image_processor(image, return_tensors='pt')['pixel_values'].half().cuda()
         # 推理
         with torch.inference_mode():

modeling_mmalaya.py

@@ -8,9 +8,9 @@ import math
 from transformers import AutoConfig, AutoModelForCausalLM
 from transformers.modeling_outputs import CausalLMOutputWithPast
 
-from mpt.modeling_mpt import MPTConfig, MPTForCausalLM, MPTModel
-from mmalaya_arch import MMAlayaMetaModel, MMAlayaMetaForCausalLM
-from configuration_mmalaya import MMAlayaMPTConfig
+from .modeling_mpt import MPTConfig, MPTForCausalLM, MPTModel
+from .mmalaya_arch import MMAlayaMetaModel, MMAlayaMetaForCausalLM
+from .configuration_mmalaya import MMAlayaMPTConfig
 
 
 class MMAlayaMPTModel(MMAlayaMetaModel, MPTModel):