fix run_model.py

This commit is contained in:
Yuki-Kokomi
2024-11-06 20:24:44 +08:00
committed by GitHub
parent 0b09f51a72
commit ea3ea7e480

View File

@@ -1,7 +1,7 @@
from tinyllava.eval.run_tiny_llava import * from tinyllava.eval.run_tiny_llava import *
import argparse import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--model_path', type=str, required=True, help="Path to the model.") parser.add_argument('--model_path', type=str, required=True, help="Path to the model.")
parser.add_argument('--src', type=str, required=True, help="Path of Input Pictures and Reference Codes.") parser.add_argument('--src', type=str, required=True, help="Path of Input Pictures and Reference Codes.")
parser.add_argument('--out', type=str, required=True, help="Any output path you like.") parser.add_argument('--out', type=str, required=True, help="Any output path you like.")
@@ -187,4 +187,4 @@ for index in range(len(input_types)):
print() print()
print("Can't Generate these inputs:") print("Can't Generate these inputs:")
print(errors) print(errors)