m1k3wn commited on
Commit
a48bfba
·
verified ·
1 Parent(s): 84f4f1d

Update app.py

Browse files

removes local_files_only caching attempt

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,13 +73,13 @@ class ModelManager:
73
  tokenizer = T5Tokenizer.from_pretrained(
74
  model_path,
75
  token=HF_TOKEN,
76
- local_files_only=True # Cache after first load
77
  )
78
 
79
  model = T5ForConditionalGeneration.from_pretrained(
80
  model_path,
81
  token=HF_TOKEN,
82
- local_files_only=True,
83
  low_cpu_mem_usage=True,
84
  torch_dtype=torch.float32
85
  )
 
73
  tokenizer = T5Tokenizer.from_pretrained(
74
  model_path,
75
  token=HF_TOKEN,
76
+ local_files_only=False # Cache after first load
77
  )
78
 
79
  model = T5ForConditionalGeneration.from_pretrained(
80
  model_path,
81
  token=HF_TOKEN,
82
+ local_files_only=False,
83
  low_cpu_mem_usage=True,
84
  torch_dtype=torch.float32
85
  )