File tree Expand file tree Collapse file tree 3 files changed +13
-1
lines changed Expand file tree Collapse file tree 3 files changed +13
-1
lines changed Original file line number Diff line number Diff line change 55
66import numpy as np
77import torch
8- from transformers import is_torchvision_available
98
109from .models import UNet2DConditionModel
1110from .utils import (
1211 convert_state_dict_to_diffusers ,
1312 convert_state_dict_to_peft ,
1413 deprecate ,
1514 is_peft_available ,
15+ is_torchvision_available ,
1616 is_transformers_available ,
1717)
1818
Original file line number Diff line number Diff line change 7575 is_torch_version ,
7676 is_torch_xla_available ,
7777 is_torchsde_available ,
78+ is_torchvision_available ,
7879 is_transformers_available ,
7980 is_transformers_version ,
8081 is_unidecode_available ,
Original file line number Diff line number Diff line change 278278except importlib_metadata .PackageNotFoundError :
279279 _peft_available = False
280280
281+ _torchvision_available = importlib .util .find_spec ("torchvision" ) is not None
282+ try :
283+ _torchvision_version = importlib_metadata .version ("torchvision" )
284+ logger .debug (f"Successfully imported torchvision version { _torchvision_version } " )
285+ except importlib_metadata .PackageNotFoundError :
286+ _torchvision_available = False
287+
281288
282289def is_torch_available ():
283290 return _torch_available
@@ -367,6 +374,10 @@ def is_peft_available():
367374 return _peft_available
368375
369376
377+ def is_torchvision_available ():
378+ return _torchvision_available
379+
380+
370381# docstyle-ignore
371382FLAX_IMPORT_ERROR = """
372383{0} requires the FLAX library but it was not found in your environment. Checkout the instructions on the
You can’t perform that action at this time.
0 commit comments