EveryonesGPT_Vision_Instruct

Running on Zero

Commit

89d297d

1 Parent(s): 5a4a0a6

fixed prompt format!

Files changed (3) hide show

__pycache__/model.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/model.cpython-310.pyc and b/__pycache__/model.cpython-310.pyc differ

__pycache__/vlm_inference.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/vlm_inference.cpython-310.pyc and b/__pycache__/vlm_inference.cpython-310.pyc differ

app.py CHANGED Viewed

@@ -68,10 +68,10 @@ def chat_fn(message, history, temperature, top_p, top_k):
         images=image.convert("RGB"),
         return_tensors="pt"
     )["pixel_values"].to(device)
-    prompt = (
-        f"{text}"
-    )
     # ★ ここが重要：累積して yield
     output = ""

         images=image.convert("RGB"),
         return_tensors="pt"
     )["pixel_values"].to(device)
+    ### NEW ###
+    prompt = f"<user>\n{text}<assistant>\n"
+    ### NEW ###
     # ★ ここが重要：累積して yield
     output = ""