saffr0n commited on
Commit
d15fd17
1 Parent(s): 06009ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ if torch.cuda.is_available():
36
  model_id = "Jacaranda/UlizaLlama"
37
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
38
  tokenizer = AutoTokenizer.from_pretrained(model_id)
39
- tokenizer.add_special_tokens({'pad_token': '[PAD]'}
40
  tokenizer.chat_template = PROMPT_TEMPLATE
41
  tokenizer.use_default_system_prompt = False
42
 
 
36
  model_id = "Jacaranda/UlizaLlama"
37
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
38
  tokenizer = AutoTokenizer.from_pretrained(model_id)
39
+ tokenizer.add_special_tokens({'pad_token': '[PAD]'})
40
  tokenizer.chat_template = PROMPT_TEMPLATE
41
  tokenizer.use_default_system_prompt = False
42