yurezsml commited on
Commit
fb59a60
1 Parent(s): a8786dc

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ model = AutoModelForCausalLM.from_pretrained(
41
  trust_remote_code=True
42
  )
43
 
44
- ft_model = PeftModel.from_pretrained(model, "yurezsml/phi2_chan", offload_dir="")
45
 
46
  def remove_accents(input_str):
47
  nfkd_form = unicodedata.normalize('NFKD', input_str)
 
41
  trust_remote_code=True
42
  )
43
 
44
+ ft_model = PeftModel.from_pretrained(model, "yurezsml/phi2_chan", offload_dir="offload/")
45
 
46
  def remove_accents(input_str):
47
  nfkd_form = unicodedata.normalize('NFKD', input_str)
requirements.txt CHANGED
@@ -7,7 +7,7 @@ datasets
7
  bitsandbytes
8
  accelerate
9
  transformers==4.38.1
10
- peft==0.9.0
11
  bs4
12
  requests
13
  nltk==3.8.1
 
7
  bitsandbytes
8
  accelerate
9
  transformers==4.38.1
10
+ peft
11
  bs4
12
  requests
13
  nltk==3.8.1