tmmdev commited on
Commit
5c43f7a
·
verified ·
1 Parent(s): fdb7559

Update pattern_analyzer.py

Browse files
Files changed (1) hide show
  1. pattern_analyzer.py +1 -1
pattern_analyzer.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
- os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
 
1
  import os
2
+ os.environ['HF_HOME'] = '/tmp/huggingface'
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5