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)
|