18 lines
599 B
Python
18 lines
599 B
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (C) 2018-2023 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
tf_hub_cache_dir = os.environ.get('TFHUB_CACHE_DIR',
|
|
os.path.join(tempfile.gettempdir(), "tfhub_modules"))
|
|
os.environ['TFHUB_CACHE_DIR'] = tf_hub_cache_dir
|
|
|
|
hf_hub_cache_dir = os.environ.get('HUGGINGFACE_HUB_CACHE',
|
|
tempfile.gettempdir())
|
|
os.environ['HUGGINGFACE_HUB_CACHE'] = hf_hub_cache_dir
|
|
|
|
# supported_devices : CPU, GPU, GNA
|
|
test_device = os.environ.get('TEST_DEVICE', 'CPU;GPU').split(';')
|