Hugging Face's logo

Spaces:
vojay
/
mental-health-demo
Runtime error

App Files Community
vojay commited on
Commit
41ce050
·
verified ·
1 Parent(s): cdf13f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -17,9 +17,7 @@ adapter_model_id = "vojay/Llama-2-7b-chat-hf-mental-health"
17
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
18
  model.load_adapter(adapter_model_id)
19
 
20
- tokenizer = AutoTokenizer.from_pretrained(model_id)
21
- tokenizer.pad_token = tokenizer.eos_token
22
- tokenizer.padding_side = "right"
23
 
24
 
25
  def get_base_prompt():
 
17
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
18
  model.load_adapter(adapter_model_id)
19
 
20
+ tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side="left")
 
 
21
 
22
 
23
  def get_base_prompt():