files

BERTmodel.py

@@ -12,7 +12,7 @@ from transformers import (
 import torch.nn.functional as F
 import os
 
-CACHE_DIR  = "./hf_cache"
+CACHE_DIR  = "/data/hf_cache"
  
     
 class BERTVisualizer(TransformerVisualizer):

DISTILLBERTmodel.py

@@ -11,7 +11,7 @@ from transformers import (
     DistilBertForMaskedLM, DistilBertForSequenceClassification
 )    
 
-CACHE_DIR  = "./hf_cache"
+CACHE_DIR  = "/data/hf_cache"
 class DistilBERTVisualizer(TransformerVisualizer):
     def __init__(self, task):
         super().__init__()

ROBERTAmodel.py

@@ -6,7 +6,9 @@ from transformers import (
     RobertaForMaskedLM, RobertaForSequenceClassification
 )
 import os
-CACHE_DIR  = "./hf_cache"
+
+CACHE_DIR  = "/data/hf_cache"
+
 class RoBERTaVisualizer(TransformerVisualizer):
     def __init__(self, task):
         super().__init__()