Upload test_mgie.py with huggingface_hub

test_mgie.py

@@ -114,6 +114,7 @@ for i in tqdm(range(1)):
     
     img = image_processor.preprocess(img, return_tensors='pt')['pixel_values'][0]
     xmb = data[i]["content"][1]["text"]
+    xb = data[i]["content"][1]["text"]
     txt = "what will this image be like if '%s'"%(txt)
     txt = txt+'\n'+DEFAULT_IM_START_TOKEN+DEFAULT_IMAGE_PATCH_TOKEN*image_token_len+DEFAULT_IM_END_TOKEN
     conv = conv_templates['vicuna_v1'].copy()
@@ -140,9 +141,9 @@ for i in tqdm(range(1)):
     res.save(save_img_path)
     token_indices = T.arange(1, len(xmb) + 1)
 
-    token_indices = get_indices_to_alter(pipe, xmb) if token_indices is None else token_indices
+    token_indices = get_indices_to_alter(pipe, xb) if token_indices is None else token_indices
     img = vis_utils.show_cross_attention(attention_store=controller,
-                                        prompt=xmb,
+                                        prompt=xb,
                                         tokenizer=tokenizer,
                                         res=16,
                                         from_where=("up", "down", "mid"),