Delete demo_test.py

demo_test.py

@@ -1,44 +0,0 @@
-# from modeling_llavanext_for_embedding import LLaVANextForEmbedding
-# from transformers import LlavaNextProcessor
-
-# model = LLaVANextForEmbedding.from_pretrained("/share/junjie/code/VISTA2/240920mllmemb/llm_dense_retriever/MMRet-release/test").cuda()
-# processor = LlavaNextProcessor.from_pretrained("/share/junjie/code/VISTA2/240920mllmemb/llm_dense_retriever/MMRet-release/test")
-
-# texts = "find a image of a dog"
-
-# inputs = processor(texts, return_tensors="pt").to("cuda")
-# outputs = model(**inputs)
-# print(outputs)
-
-
-import torch
-from transformers import LlavaNextProcessor, AutoModel
-
-model = AutoModel.from_pretrained("/share/junjie/code/VISTA2/240920mllmemb/llm_dense_retriever/MMRet-release/MMRet-MLLM", trust_remote_code=True).cuda()
-model = model.eval()
-processor = LlavaNextProcessor.from_pretrained("/share/junjie/code/VISTA2/240920mllmemb/llm_dense_retriever/MMRet-release/MMRet-MLLM")
-
-texts = "[INST] \n <instruct>  <query> find a image of a dog \n [/INST]"
-
-inputs = processor(texts, return_tensors="pt").to("cuda")
-outputs = model(**inputs)[:, -1, :]
-embeddings = torch.nn.functional.normalize(outputs, dim=-1)
-
-print(embeddings)
-
-
-
-from transformers import LlavaNextProcessor, AutoModel
-import torch
-
-model_name = "/share/junjie/code/VISTA2/240920mllmemb/llm_dense_retriever/MMRet-release/MMRet-MLLM"
-model = AutoModel.from_pretrained(model_name, trust_remote_code=True).cuda()
-model = model.eval()
-model.set_processor(model_name)
-inputs = model.data_process(text="find a image of a dog", q_or_c="query")
-
-model_output = model(**inputs, output_hidden_states=True)
-embeddings = model_output[:, -1, :]
-embeddings = torch.nn.functional.normalize(embeddings, dim=-1)
-
-print(embeddings)