From 31b0c5bdda3e9bb188b8ed249bf5b9dabef34c1e Mon Sep 17 00:00:00 2001
From: yilun_jin <jyl.jal123@gmail.com>
Date: Fri, 22 Mar 2024 05:08:06 +0000
Subject: [PATCH] Update models/user_config.py

---
 models/user_config.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/models/user_config.py b/models/user_config.py
index c152a6b..fc0e284 100644
--- a/models/user_config.py
+++ b/models/user_config.py
@@ -1,12 +1,13 @@
 # Importing DummyModel from the models package.
 # The DummyModel class is located in the dummy_model.py file inside the 'models' directory.
-from models.dummy_model import DummyModel
+from models.dummy_model import DummyModel, Vicuna2ZeroShot
 
 # This line establishes an alias for the DummyModel class to be used within this script.
 # Instead of directly using DummyModel everywhere in the code, we're assigning it to 'UserModel'.
 # This approach allows for easier reference to your model class when evaluating your models,
 UserModel = DummyModel
 
+
 # When implementing your own model please follow this pattern:
 #
 # from models.your_model import YourModel
@@ -17,3 +18,6 @@ UserModel = DummyModel
 # Finally, assign YourModel to UserModel as shown below to use it throughout your script.
 #
 # UserModel = YourModel
+
+# For example, you can try the Vicuna2ZeroShot baseline by uncommenting the following line. 
+# UserModel = Vicuna2ZeroShot
-- 
GitLab