Update my_model/config/kbvqa_config.py
Browse files
my_model/config/kbvqa_config.py
CHANGED
|
@@ -7,6 +7,7 @@ MAX_CONTEXT_WINDOW = 4000
|
|
| 7 |
ADD_EOS_TOKEN = False
|
| 8 |
TRUST_REMOTE = False
|
| 9 |
USE_FAST = True
|
|
|
|
| 10 |
|
| 11 |
# Access Token
|
| 12 |
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
|
|
|
|
| 7 |
ADD_EOS_TOKEN = False
|
| 8 |
TRUST_REMOTE = False
|
| 9 |
USE_FAST = True
|
| 10 |
+
LOW_CPU_MEM_USAGE = True
|
| 11 |
|
| 12 |
# Access Token
|
| 13 |
HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
|