Spaces:
Sleeping
Sleeping
research14
commited on
Commit
•
93184b1
1
Parent(s):
5d4d985
updt runllm
Browse files- run_llm.py +6 -0
run_llm.py
CHANGED
@@ -464,8 +464,14 @@ def run_llm_interface(model_path, prompt, sentence):
|
|
464 |
)
|
465 |
|
466 |
# Run the main function
|
|
|
|
|
467 |
main(args=args)
|
468 |
|
|
|
|
|
|
|
|
|
469 |
if __name__ == "__main__":
|
470 |
parser = argparse.ArgumentParser()
|
471 |
add_model_args(parser)
|
|
|
464 |
)
|
465 |
|
466 |
# Run the main function
|
467 |
+
# For simplicity, assuming prompt values 1, 2, and 3 correspond to different strategies
|
468 |
+
# You may need to adjust this based on your actual logic
|
469 |
main(args=args)
|
470 |
|
471 |
+
# Return dummy values for now, replace with actual outputs
|
472 |
+
return "Strategy 1 Output", "Strategy 2 Output", "Strategy 3 Output"
|
473 |
+
|
474 |
+
|
475 |
if __name__ == "__main__":
|
476 |
parser = argparse.ArgumentParser()
|
477 |
add_model_args(parser)
|