ToletiSri commited on
Commit
2dcf2f9
1 Parent(s): 6c66946

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ model = AutoModelForCausalLM.from_pretrained(
23
  trust_remote_code=True
24
  )
25
  model.config.use_cache = False
26
- model.load_adapter("checkpoint")
27
 
28
  tokenizer = AutoTokenizer.from_pretrained("checkpoint_500", trust_remote_code=True)
29
  tokenizer.pad_token = tokenizer.eos_token
 
23
  trust_remote_code=True
24
  )
25
  model.config.use_cache = False
26
+ model.load_adapter("checkpoint_500")
27
 
28
  tokenizer = AutoTokenizer.from_pretrained("checkpoint_500", trust_remote_code=True)
29
  tokenizer.pad_token = tokenizer.eos_token