No1r97 commited on
Commit
8d31bb0
1 Parent(s): a54d7cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import re
2
  import time
3
  import json
@@ -10,6 +11,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
10
  from peft import PeftModel
11
  from datetime import date, datetime, timedelta
12
 
 
13
 
14
  access_token = "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"
15
 
 
1
+ import os
2
  import re
3
  import time
4
  import json
 
11
  from peft import PeftModel
12
  from datetime import date, datetime, timedelta
13
 
14
+ os.environ['HF_HOME'] = '/data/.huggingface'
15
 
16
  access_token = "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"
17