Add os

neuralmagic · Jul 21, 2023 · b741348 · b741348
1 parent 0d7d677
commit b741348
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/deepsparse/transformers/pipelines/text_generation.py b/src/deepsparse/transformers/pipelines/text_generation.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import logging
+import os
 import warnings
 from typing import List, Optional, Tuple, Type, Union