AurelioAguirre commited on
Commit
ff4bec2
·
1 Parent(s): 0776594

moved folders v3

Browse files
Files changed (1) hide show
  1. main/api.py +1 -1
main/api.py CHANGED
@@ -42,7 +42,7 @@ class LLMApi:
42
  self.cache_path.mkdir(parents=True, exist_ok=True)
43
 
44
  # Set cache directory for transformers
45
- os.environ['TRANSFORMERS_CACHE'] = str(self.cache_path)
46
 
47
  self.logger.info("LLM API initialized successfully")
48
 
 
42
  self.cache_path.mkdir(parents=True, exist_ok=True)
43
 
44
  # Set cache directory for transformers
45
+ os.environ['HF_HOME'] = str(self.cache_path)
46
 
47
  self.logger.info("LLM API initialized successfully")
48