Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
8600d7f
1
Parent(s):
0c94688
update
Browse files
embodied_gen/utils/gpt_clients.py
CHANGED
@@ -61,7 +61,6 @@ class GPTclient:
|
|
61 |
|
62 |
self.endpoint = endpoint
|
63 |
self.model_name = model_name
|
64 |
-
self.api_version = api_version
|
65 |
self.image_formats = {".png", ".jpg", ".jpeg", ".webp", ".bmp", ".gif"}
|
66 |
self.verbose = verbose
|
67 |
logger.info(f"Using GPT model: {self.model_name}.")
|
|
|
61 |
|
62 |
self.endpoint = endpoint
|
63 |
self.model_name = model_name
|
|
|
64 |
self.image_formats = {".png", ".jpg", ".jpeg", ".webp", ".bmp", ".gif"}
|
65 |
self.verbose = verbose
|
66 |
logger.info(f"Using GPT model: {self.model_name}.")
|
embodied_gen/utils/gpt_config.yaml
CHANGED
@@ -9,6 +9,6 @@ gpt-4o:
|
|
9 |
|
10 |
qwen2.5-vl:
|
11 |
endpoint: https://openrouter.ai/api/v1
|
12 |
-
api_key: sk-or-v1-
|
13 |
api_version: null
|
14 |
model_name: qwen/qwen2.5-vl-72b-instruct:free
|
|
|
9 |
|
10 |
qwen2.5-vl:
|
11 |
endpoint: https://openrouter.ai/api/v1
|
12 |
+
api_key: sk-or-v1-xxx
|
13 |
api_version: null
|
14 |
model_name: qwen/qwen2.5-vl-72b-instruct:free
|
embodied_gen/validators/urdf_convertor.py
CHANGED
@@ -366,9 +366,6 @@ class URDFGenerator(object):
|
|
366 |
image_path = combine_images_to_base64(image_path)
|
367 |
|
368 |
response = self.gpt_client.query(text_prompt, image_path)
|
369 |
-
print("text_prompt: ", text_prompt)
|
370 |
-
print("image_path: ", image_path)
|
371 |
-
print("response: ", response)
|
372 |
if response is None:
|
373 |
asset_attrs = {
|
374 |
"category": category.lower(),
|
|
|
366 |
image_path = combine_images_to_base64(image_path)
|
367 |
|
368 |
response = self.gpt_client.query(text_prompt, image_path)
|
|
|
|
|
|
|
369 |
if response is None:
|
370 |
asset_attrs = {
|
371 |
"category": category.lower(),
|