Pudding48 commited on
Commit
f62a847
·
verified ·
1 Parent(s): 6871744

Update prepare_vector_dp.py

Browse files
Files changed (1) hide show
  1. prepare_vector_dp.py +3 -0
prepare_vector_dp.py CHANGED
@@ -8,6 +8,9 @@ from huggingface_hub import hf_hub_download
8
  # from llama_cpp import Llama
9
  import os
10
 
 
 
 
11
  # model_file = Llama.from_pretrained(
12
  # repo_id="Pudding48/TinyLLamaTest",
13
  # filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",
 
8
  # from llama_cpp import Llama
9
  import os
10
 
11
+ # Fix Hugging Face cache path for Docker Spaces
12
+ os.environ["HF_HOME"] = "/home/user/app/hf_cache"
13
+
14
  # model_file = Llama.from_pretrained(
15
  # repo_id="Pudding48/TinyLLamaTest",
16
  # filename="tinyllama-1.1b-chat-v1.0.Q8_0.gguf",