STC-demo-v2_dev

Sleeping

fCola commited on May 21

Commit

88e33a6

verified ·

1 Parent(s): 981bdb7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ class HfModelWrapper:
         self.tokenizer = AutoTokenizer.from_pretrained(model_path)
         self.sys_prompt = sys_prompt
         self.adapter_path = adapter_path
-        self.model.load_adapter(self.adapter_path, token=token)
         self.model.enable_adapters()
     def build_prompt(self, user_msg, history):

         self.tokenizer = AutoTokenizer.from_pretrained(model_path)
         self.sys_prompt = sys_prompt
         self.adapter_path = adapter_path
+        self.model.load_adapter(self.adapter_path, token=hf_token)
         self.model.enable_adapters()
     def build_prompt(self, user_msg, history):