sft
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ disable_btn = gr.Button(interactive=False)
|
|
33 |
argparser = argparse.ArgumentParser()
|
34 |
argparser.add_argument("--server_name", default="0.0.0.0", type=str)
|
35 |
argparser.add_argument("--port", default="6324", type=str)
|
36 |
-
argparser.add_argument("--model-path", default="umd-vt-nyu/clip-evaclip-und-gen-
|
37 |
argparser.add_argument("--model-base", type=str, default=None)
|
38 |
argparser.add_argument("--num-gpus", type=int, default=1)
|
39 |
argparser.add_argument("--conv-mode", type=str, default="plain")
|
@@ -121,7 +121,6 @@ def generate(state, imagebox, textbox, image_process_mode, temperature, top_p, m
|
|
121 |
|
122 |
#images = [load_image_from_base64(image) for image in images]
|
123 |
image_sizes = [image.size for image in images]
|
124 |
-
print(images, image_processor, model.config)
|
125 |
|
126 |
images = process_images(images, image_processor, model.config)
|
127 |
|
|
|
33 |
argparser = argparse.ArgumentParser()
|
34 |
argparser.add_argument("--server_name", default="0.0.0.0", type=str)
|
35 |
argparser.add_argument("--port", default="6324", type=str)
|
36 |
+
argparser.add_argument("--model-path", default="umd-vt-nyu/clip-evaclip-und-gen-sft-3v", type=str)
|
37 |
argparser.add_argument("--model-base", type=str, default=None)
|
38 |
argparser.add_argument("--num-gpus", type=int, default=1)
|
39 |
argparser.add_argument("--conv-mode", type=str, default="plain")
|
|
|
121 |
|
122 |
#images = [load_image_from_base64(image) for image in images]
|
123 |
image_sizes = [image.size for image in images]
|
|
|
124 |
|
125 |
images = process_images(images, image_processor, model.config)
|
126 |
|