vicuna

utils/model.py

@@ -4,7 +4,7 @@ import torch
 
 
 class Model():
-    def __init__(self, model="tiiuae/falcon-7b-instruct") -> None:
+    def __init__(self, model="lmsys/vicuna-7b-v1.5") -> None:
         pass
         self.tokenizer = AutoTokenizer.from_pretrained(model)
         self.pipeline = transformers.pipeline(