File tree Expand file tree Collapse file tree 3 files changed +2
-6
lines changed Expand file tree Collapse file tree 3 files changed +2
-6
lines changed Original file line number Diff line number Diff line change 1616
1717import logging
1818import os
19- import tempfile
2019import shutil
2120from abc import ABC , abstractmethod
2221from pathlib import Path
@@ -120,12 +119,12 @@ def _cleanup_temp_resources(self):
120119 # It's a TemporaryDirectory object
121120 logging .info (f"Cleaning up temporary directory: { self ._temp_dir .name } " )
122121 self ._temp_dir .cleanup ()
123- logging .info (f "Temporary directory cleanup completed" )
122+ logging .info ("Temporary directory cleanup completed" )
124123 elif isinstance (self ._temp_dir , (str , Path )):
125124 # It's a path
126125 logging .info (f"Cleaning up temporary path: { self ._temp_dir } " )
127126 shutil .rmtree (self ._temp_dir , ignore_errors = True )
128- logging .info (f "Temporary path cleanup completed" )
127+ logging .info ("Temporary path cleanup completed" )
129128 except Exception as e :
130129 # Log cleanup errors for debugging
131130 logging .warning (f"Error during temp directory cleanup: { e } " )
Original file line number Diff line number Diff line change 2828)
2929from transformers .generation .configuration_utils import GenerationConfig
3030
31- from executorch import version as executorch_version
3231from optimum .executorch .attentions .custom_sdpa import get_custom_sdpa_for_ring_kv_cache
3332from optimum .utils .import_utils import is_transformers_version
3433
Original file line number Diff line number Diff line change 1616from typing import Optional
1717
1818import torch
19- import torchao
20- from packaging .version import parse
2119
2220
2321def quantize_model_ (
You can’t perform that action at this time.
0 commit comments