Spaces:
Sleeping
Sleeping
Update app.py
Browse filesremoves local_files_only caching attempt
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=
|
77 |
)
|
78 |
|
79 |
model = T5ForConditionalGeneration.from_pretrained(
|
80 |
model_path,
|
81 |
token=HF_TOKEN,
|
82 |
-
local_files_only=
|
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 |
)
|