diff --git a/models/user_config.py b/models/user_config.py index c152a6b3ab2b74ab1194bca06b7408161bec82f3..fc0e2842cf66646ddfc18031d5dd809de6b81cad 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