Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- inference.py +3 -3
inference.py
CHANGED
@@ -6,10 +6,10 @@ from safetensors.torch import load_file
|
|
6 |
from datasets import load_dataset
|
7 |
from pprintpp import pprint
|
8 |
|
9 |
-
import os
|
10 |
|
11 |
-
# Đặt biến môi trường HF_HOME
|
12 |
-
os.environ["HF_HOME"] = "/tmp/hf_cache"
|
13 |
|
14 |
class InferenceModel:
|
15 |
"""
|
|
|
6 |
from datasets import load_dataset
|
7 |
from pprintpp import pprint
|
8 |
|
9 |
+
# import os
|
10 |
|
11 |
+
# # Đặt biến môi trường HF_HOME
|
12 |
+
# os.environ["HF_HOME"] = "/tmp/hf_cache"
|
13 |
|
14 |
class InferenceModel:
|
15 |
"""
|