Hugging Face's logo

Spaces:
Vasudevakrishna
/
Qlora_Phi2_Model
Sleeping

App Files Community
Vasudevakrishna commited on
Commit
4a9df9f
·
verified ·
1 Parent(s): 8b5b7ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model.config.use_cache = False
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
13
  tokenizer.pad_token = tokenizer.eos_token
14
 
15
- peft_model_folder = 'ckpts'
16
  model.load_adapter(peft_model_folder)
17
 
18
  def generate_text(input_text):
 
12
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
13
  tokenizer.pad_token = tokenizer.eos_token
14
 
15
+ peft_model_folder = './ckpts'
16
  model.load_adapter(peft_model_folder)
17
 
18
  def generate_text(input_text):