zizytd commited on
Commit
58b797f
·
verified ·
1 Parent(s): 21b2018

updating app.py v3

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ from markdown import Markdown
22
  from IPython.display import display
23
  from typing import List, Dict
24
 
25
- os.environ["TRANSFORMERS_CACHE"] = "/tmp"
26
  model_name = "ethicsadvisorproject/Llama-2-7b-ethical-chat-finetune"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp/offload")
28
  model = AutoModelForCausalLM.from_pretrained(
 
22
  from IPython.display import display
23
  from typing import List, Dict
24
 
25
+ os.environ["HF_HOME"] = "/tmp"
26
  model_name = "ethicsadvisorproject/Llama-2-7b-ethical-chat-finetune"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir="/tmp/offload")
28
  model = AutoModelForCausalLM.from_pretrained(