Update gme_inference.py
Browse files- 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 |
-
|
308 |
-
'
|
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("
|
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)
|