izhx commited on
Commit
5354088
1 Parent(s): 22270be

Update gme_inference.py

Browse files
Files changed (1) hide show
  1. gme_inference.py +3 -5
gme_inference.py CHANGED
@@ -304,13 +304,11 @@ if __name__ == '__main__':
304
  "The Tesla Cybertruck is a battery electric pickup truck built by Tesla, Inc. since 2023."
305
  ]
306
  images = [
307
- # 'https://en.wikipedia.org/wiki/File:Tesla_Cybertruck_damaged_window.jpg',
308
- '/nas-alinlp/linzhang.zx/gme_space/assets/Tesla_Cybertruck_damaged_window.jpg',
309
- # 'https://en.wikipedia.org/wiki/File:2024_Tesla_Cybertruck_Foundation_Series,_front_left_(Greenwich).jpg',
310
- '/nas-alinlp/linzhang.zx/gme_space/assets/2024_Tesla_Cybertruck_Foundation_Series,_front_left_(Greenwich).jpg',
311
  ]
312
 
313
- gme = GmeQwen2VL("/nas-alinlp/linzhang.zx/gme_space/gme-Qwen2-VL-2B-instruct")
314
 
315
  # Single-modal embedding
316
  e_text = gme.get_text_embeddings(texts=texts)
 
304
  "The Tesla Cybertruck is a battery electric pickup truck built by Tesla, Inc. since 2023."
305
  ]
306
  images = [
307
+ 'https://en.wikipedia.org/wiki/File:Tesla_Cybertruck_damaged_window.jpg',
308
+ 'https://en.wikipedia.org/wiki/File:2024_Tesla_Cybertruck_Foundation_Series,_front_left_(Greenwich).jpg',
 
 
309
  ]
310
 
311
+ gme = GmeQwen2VL("Alibaba-NLP/gme-Qwen2-VL-2B-Instruct")
312
 
313
  # Single-modal embedding
314
  e_text = gme.get_text_embeddings(texts=texts)