File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree 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