diffusers 0.30.0__py3-none-any.whl → 0.30.2__py3-none-any.whl

diffusers/utils/import_utils.py

@@ -330,6 +330,15 @@ except importlib_metadata.PackageNotFoundError:
 
 _is_google_colab = "google.colab" in sys.modules or any(k.startswith("COLAB_") for k in os.environ)
 
+_imageio_available = importlib.util.find_spec("imageio") is not None
+if _imageio_available:
+    try:
+        _imageio_version = importlib_metadata.version("imageio")
+        logger.debug(f"Successfully imported imageio version {_imageio_version}")
+
+    except importlib_metadata.PackageNotFoundError:
+        _imageio_available = False
+
 
 def is_torch_available():
     return _torch_available
@@ -447,6 +456,10 @@ def is_sentencepiece_available():
     return _sentencepiece_available
 
 
+def is_imageio_available():
+    return _imageio_available
+
+
 # docstyle-ignore
 FLAX_IMPORT_ERROR = """
 {0} requires the FLAX library but it was not found in your environment. Checkout the instructions on the
@@ -575,6 +588,11 @@ BITSANDBYTES_IMPORT_ERROR = """
 {0} requires the bitsandbytes library but it was not found in your environment. You can install it with pip: `pip install bitsandbytes`
 """
 
+# docstyle-ignore
+IMAGEIO_IMPORT_ERROR = """
+{0} requires the imageio library and ffmpeg but it was not found in your environment. You can install it with pip: `pip install imageio imageio-ffmpeg`
+"""
+
 BACKENDS_MAPPING = OrderedDict(
     [
         ("bs4", (is_bs4_available, BS4_IMPORT_ERROR)),
@@ -599,6 +617,7 @@ BACKENDS_MAPPING = OrderedDict(
         ("safetensors", (is_safetensors_available, SAFETENSORS_IMPORT_ERROR)),
         ("bitsandbytes", (is_bitsandbytes_available, BITSANDBYTES_IMPORT_ERROR)),
         ("sentencepiece", (is_sentencepiece_available, SENTENCEPIECE_IMPORT_ERROR)),
+        ("imageio", (is_imageio_available, IMAGEIO_IMPORT_ERROR)),
     ]
 )
 

diffusers/utils/loading_utils.py

@@ -6,7 +6,7 @@ import PIL.Image
 import PIL.ImageOps
 import requests
 
-from .import_utils import BACKENDS_MAPPING, is_opencv_available
+from .import_utils import BACKENDS_MAPPING, is_imageio_available
 
 
 def load_image(
@@ -81,7 +81,8 @@ def load_video(
 
     if is_url:
         video_data = requests.get(video, stream=True).raw
-        video_path = tempfile.NamedTemporaryFile(suffix=os.path.splitext(video)[1], delete=False).name
+        suffix = os.path.splitext(video)[1] or ".mp4"
+        video_path = tempfile.NamedTemporaryFile(suffix=suffix, delete=False).name
         was_tempfile_created = True
         with open(video_path, "wb") as f:
             f.write(video_data.read())
@@ -99,19 +100,22 @@ def load_video(
             pass
 
     else:
-        if is_opencv_available():
-            import cv2
+        if is_imageio_available():
+            import imageio
         else:
-            raise ImportError(BACKENDS_MAPPING["opencv"][1].format("load_video"))
+            raise ImportError(BACKENDS_MAPPING["imageio"][1].format("load_video"))
 
-        video_capture = cv2.VideoCapture(video)
-        success, frame = video_capture.read()
-        while success:
-            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            pil_images.append(PIL.Image.fromarray(frame))
-            success, frame = video_capture.read()
+        try:
+            imageio.plugins.ffmpeg.get_exe()
+        except AttributeError:
+            raise AttributeError(
+                "`Unable to find an ffmpeg installation on your machine. Please install via `pip install imageio-ffmpeg"
+            )
 
-        video_capture.release()
+        with imageio.get_reader(video) as reader:
+            # Read all frames
+            for frame in reader:
+                pil_images.append(PIL.Image.fromarray(frame))
 
     if was_tempfile_created:
         os.remove(video_path)

{diffusers-0.30.0.dist-info → diffusers-0.30.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffusers
-Version: 0.30.0
+Version: 0.30.2
 Summary: State-of-the-art diffusion in PyTorch and JAX.
 Home-page: https://github.com/huggingface/diffusers
 Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/diffusers/graphs/contributors)

{diffusers-0.30.0.dist-info → diffusers-0.30.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-diffusers/__init__.py,sha256=vQytNtLHin9OJg3IjGwms3iRGNncaloWFJvq34O4v-c,34105
+diffusers/__init__.py,sha256=41QUjRiVqQrTK2FPRwBSz59s8TYEbRSNB54bmuQSUO0,34105
 diffusers/callbacks.py,sha256=m8ariuJC-WaPHMZn1zUyXG8hlAvaOvEW_6YWdKo--eo,6717
 diffusers/configuration_utils.py,sha256=yJiYuhgyHRK3AgjL15mevGzBJg7GsXZOYVjWVgy4AIs,32484
 diffusers/dependency_versions_check.py,sha256=J_ZAEhVN6uLWAOUZCJrcGJ7PYxUek4f_nwGTFM7LTk8,1271
@@ -16,14 +16,14 @@ diffusers/experimental/__init__.py,sha256=0C9ExG0XYiGZuzFJkZuJ53K6Ix5ylF2kWe4PGA
 diffusers/experimental/rl/__init__.py,sha256=Gcoznw9rYjfMvswH0seXekKYDAAN1YXXxZ-RWMdzvrE,57
 diffusers/experimental/rl/value_guided_sampling.py,sha256=gnUDVNx5nIVJDWxhHBlga4j7VQTxSTkUI1QaCnpiWAM,6033
 diffusers/loaders/__init__.py,sha256=J3fmn2UoRVDh_QaeNMH-Zwn7ejwVBmTWofla6Tp6Rek,4005
-diffusers/loaders/ip_adapter.py,sha256=U7ReoJQV2ZHL46URAw8JertFT9PE0z6EY-jnI1FIumo,16966
+diffusers/loaders/ip_adapter.py,sha256=cdQGSWcJECas_Pz1nNMhoSBceA9zOXFEvkAvVdq251o,17078
 diffusers/loaders/lora_base.py,sha256=ni68uQ1apLAjiwNodXv-teSgVlQFmKIiMhU3nAHoYyo,31716
 diffusers/loaders/lora_conversion_utils.py,sha256=tbiyxd9ByjGDBv2fSBOo8uPUh0uProjZBbXFH3zoG7A,14322
-diffusers/loaders/lora_pipeline.py,sha256=odT-DSh68l7ObLt_jZvMQ7G-Qej42TryHY9xO8MI-J0,104485
+diffusers/loaders/lora_pipeline.py,sha256=3ZREkA-aVGLZyVaDoP_HzlRVYYU4C-7-uNgG5EFtfcM,106023
 diffusers/loaders/peft.py,sha256=EtOFz7ezjBVwQrwIZ0z-bwWGIWKAroWh0oCpa_mu2pw,16127
-diffusers/loaders/single_file.py,sha256=HiSw12-36GOP-YtWkVqWsd4JxM8EiaU7NhWYQjxO7ho,24149
+diffusers/loaders/single_file.py,sha256=HD5p15rgNx6Ao3G8ply3Uq7sS7SaKl0j8wKj2vHIJPA,24222
 diffusers/loaders/single_file_model.py,sha256=CiAXlpS3S9EzN6X8tDE4hzjN5fxu53wS_e-BLSmZ6U8,14330
-diffusers/loaders/single_file_utils.py,sha256=iGjjwHRpsbHKiy0GbR6HSsw7mv9GZlkA4D6xY1bAKUI,88875
+diffusers/loaders/single_file_utils.py,sha256=F4WO_d9RKY6Dcqn6LqNGvziahZr4A6RJzGeM_2tEAkU,89987
 diffusers/loaders/textual_inversion.py,sha256=QjRUJZfyUJFY-sYAIVLUZgNuuk65xlbiIwJ3CD_0rAo,26655
 diffusers/loaders/unet.py,sha256=thoMcf4Q2F5yn1UJVFEAf64F5TgZTULrmrcFHkZeJ4I,43709
 diffusers/loaders/unet_loader_utils.py,sha256=9IHd_RlKqMstSO8G7btUdL1-Y3-fGX7Kbc4frEbRVVM,6220
@@ -33,7 +33,7 @@ diffusers/models/activations.py,sha256=2BXSp2kNk8-WZKwMFzVZKJqyDc8NhHv2yEdHKwhg3
 diffusers/models/adapter.py,sha256=XuVoUbhLkPEOeClqYiTm8KGMQrXgn2WZU3I73-jkcew,24723
 diffusers/models/attention.py,sha256=bFYlFf46Fo6R6bs1zvTLqvjcDeb5XmZ5rInB1kmp1-U,50270
 diffusers/models/attention_flax.py,sha256=Ju2KJCqsx_LIiz0i1pBcek7RMKTmVOIF4SvEcOqgJ1c,20250
-diffusers/models/attention_processor.py,sha256=eWxrwyl3orXrPgNcXnSi4foRXJk4iAYFgeR-RrCrZLE,176917
+diffusers/models/attention_processor.py,sha256=r-R_9tM8hlr3ReKO7UvYBItm_VLefCs7rTeGfoJndsk,182643
 diffusers/models/controlnet.py,sha256=XcKsqCLtFbbvdJsbYjzxRUblmWYNMeo0kSP54d_m4yk,43249
 diffusers/models/controlnet_flax.py,sha256=_UuB-tNxQ9eR8v3dqDhF2Mz-6thIdEI6BlY8BpWpkvU,16710
 diffusers/models/controlnet_hunyuan.py,sha256=cb6SvFyiqHfkWqDiyfQjyZehZJT19JeZlyVGdZnPE7c,16913
@@ -41,7 +41,7 @@ diffusers/models/controlnet_sd3.py,sha256=NO7w6FT3r_JINbzojUcModp4oukZGzjV6a0n0s
 diffusers/models/controlnet_sparsectrl.py,sha256=V5wnisfoQQS4-9GVWb3E5DSaPjVXoJd-gAgwu_7vPEg,38539
 diffusers/models/controlnet_xs.py,sha256=zhz1O1ARvm1VhxHtVpZap7_I_8xq21RdsGGvraU9NJI,86865
 diffusers/models/downsampling.py,sha256=Li03bTffSRQVJtN41W7gPamEEkgymwX9fP_kICzGTO4,15828
-diffusers/models/embeddings.py,sha256=a_I8WdjBKIiG51VRrbCUmd0sY2VBSRc3GSBkvZEqBa4,60815
+diffusers/models/embeddings.py,sha256=CXu4qvaCSg91R-YUATsJXyC5_pDXXTABOjuTG0rYUhY,64429
 diffusers/models/embeddings_flax.py,sha256=hNN63qtv9Ht6VEeidrIJdrdKWm0o8mumcgzxMNaR2H0,3445
 diffusers/models/lora.py,sha256=7LbI7bj8yk9GptoOnOLrhzarFcVSQX47LuGoZ1MBK0A,18829
 diffusers/models/model_loading_utils.py,sha256=YPp7aPymfOJ1Yd6oLgN3zVEmCPhqx3k6pMi9-1Hftnc,8534
@@ -59,7 +59,7 @@ diffusers/models/vq_model.py,sha256=ZUGlHjd0mhhfy2mtOIsMg4hIvUUL9j8MB2gQDVWEkQU,
 diffusers/models/autoencoders/__init__.py,sha256=1S4H1g51svAn9RfivWEmctE5d95Sr99cSZQlLggmzHQ,421
 diffusers/models/autoencoders/autoencoder_asym_kl.py,sha256=BQcgLKgQY3D9dBoqcb6G1Syvh02mE548z2PBSNbOKTI,7720
 diffusers/models/autoencoders/autoencoder_kl.py,sha256=3sjWEQuNDN_yeGJBEoEt_VCHwYBvcgD6LWBcPLL3L7c,22305
-diffusers/models/autoencoders/autoencoder_kl_cogvideox.py,sha256=gCsnXq0VhIZWtnjoesE6X9vzUuWSP2fxWc6H8JpG3jo,40324
+diffusers/models/autoencoders/autoencoder_kl_cogvideox.py,sha256=--8SEoDVTivctb7_vv2oP5kLiXtuWSKwo0w5bljZTh8,51855
 diffusers/models/autoencoders/autoencoder_kl_temporal_decoder.py,sha256=ZiQyVdCrxG2deXe7hdU9CAJ4Y6uAWAkWCXiOsDsagHQ,16248
 diffusers/models/autoencoders/autoencoder_oobleck.py,sha256=gDEs-MUaCvuJqPs9upiRg8cs2vWFY7wOTvUHfFC2TEU,17046
 diffusers/models/autoencoders/autoencoder_tiny.py,sha256=7wglKu1FYGgnWHdHeVu8N_OxerqImQWCoyaVwlQzrZ8,15992
@@ -67,8 +67,8 @@ diffusers/models/autoencoders/consistency_decoder_vae.py,sha256=Be_o13r7odf6wgoc
 diffusers/models/autoencoders/vae.py,sha256=Xzj7pt2H2D5AnILpzjwCyx29B70RqK-muZZBdafE6ZE,36545
 diffusers/models/autoencoders/vq_model.py,sha256=VXU4jgIIH-60mIeZB6t8oa1F9AvMSCLo-nv-Ee2Ec4Q,7812
 diffusers/models/transformers/__init__.py,sha256=DPueInTLFJNWNLnRxt4119QrDusQlEVC2QV3IUwIJ_o,953
-diffusers/models/transformers/auraflow_transformer_2d.py,sha256=1rgyxpN68XLGCs8c2ewSFMlAy4OnzOxEBNIdJ8hBmhg,21798
-diffusers/models/transformers/cogvideox_transformer_3d.py,sha256=tATWmnV4bHGmue_Q4_HtVhBl1P3dBFFeyWXlrskuYbA,15027
+diffusers/models/transformers/auraflow_transformer_2d.py,sha256=-Rk6U95hQC6EgVzRhH_33JmjODcj4Jfng6AG7n9Hm6U,22754
+diffusers/models/transformers/cogvideox_transformer_3d.py,sha256=wZzcmL8KC7AHOaUMvUZX2XpM8QSNEbaN3mCmSfiqQmA,20898
 diffusers/models/transformers/dit_transformer_2d.py,sha256=o3MT0w2m4BrdGp0KUonm9QqhVpNVzYzRyKuuPnH02Nc,11158
 diffusers/models/transformers/dual_transformer_2d.py,sha256=TEgdpVW9itJ97YgslzKnYWg-2V4Oq7AMHipMxON-72Y,7711
 diffusers/models/transformers/hunyuan_transformer_2d.py,sha256=CUzxabgvxLrJ0aOzgd7JEKwP9nK7TbZ3SJ-LHkDZ6Q4,24234
@@ -130,7 +130,7 @@ diffusers/pipelines/blip_diffusion/modeling_blip2.py,sha256=IvMpb_5bl8PKmxL11-l4
 diffusers/pipelines/blip_diffusion/modeling_ctx_clip.py,sha256=7T17m5qR9m6XUEVuOlbptERKR_b4a5lt8PvslJPUy-c,9002
 diffusers/pipelines/blip_diffusion/pipeline_blip_diffusion.py,sha256=kBof-UrMvpKaANzu_5pPYOgr56ZajgJA-XdzY9AAcHo,15001
 diffusers/pipelines/cogvideo/__init__.py,sha256=jNAtMQxLZB0Xw0w3S9s4BgqFtFERiwNlBEvpVCXFmqw,1298
-diffusers/pipelines/cogvideo/pipeline_cogvideox.py,sha256=LqY7ym2_lmz-33prOZBkXlrW3Hcru0kYbQJf7JahmJY,33531
+diffusers/pipelines/cogvideo/pipeline_cogvideox.py,sha256=lK5hGv31m9AsbZlQyGgSXJI-tXgPwodOHd2i4PpKkPA,35930
 diffusers/pipelines/consistency_models/__init__.py,sha256=q_nrLK9DH0_kLcLmRIvgvLP-vDVwloC3lBus776596c,484
 diffusers/pipelines/consistency_models/pipeline_consistency_models.py,sha256=EDnYrtJ-CGVgr1bjRSaCzAx2h_tIGdF47BSZt3HvYtQ,12368
 diffusers/pipelines/controlnet/__init__.py,sha256=V5lvvD6DALNY3InsdwVmRz65_ZlWM-mPoATmu-yxCtk,3483
@@ -209,7 +209,7 @@ diffusers/pipelines/deprecated/vq_diffusion/pipeline_vq_diffusion.py,sha256=YSYu
 diffusers/pipelines/dit/__init__.py,sha256=w6yUFMbGzaUGPKpLfEfvHlYmrKD0UErczwsHDaDtLuQ,408
 diffusers/pipelines/dit/pipeline_dit.py,sha256=hrLURHV8-ujtNF6t-nrYXVfARyVKh-_OfO9jAtaWUS0,9949
 diffusers/pipelines/flux/__init__.py,sha256=hiwnaba-laO3e2Wgh9dyzMWowHDGhghuQc5ZJKjCg3U,1457
-diffusers/pipelines/flux/pipeline_flux.py,sha256=ho1ER7DDNmGzdl1iQ0DjcXMkAgh6PWpvjSdjHkepf98,34420
+diffusers/pipelines/flux/pipeline_flux.py,sha256=rvUztMJh55AQYzrIDPOdbic6asc9da3HJ_iZBYoI-Mg,34417
 diffusers/pipelines/flux/pipeline_output.py,sha256=zKVpr6EHsvT7ADxNkvdZLv9519JPMThZEwxj9bvx-8g,598
 diffusers/pipelines/hunyuandit/__init__.py,sha256=Zby0yEsLNAoa4cf6W92QXIzyGoijI54xXRVhmrHGHsc,1302
 diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py,sha256=eq5LVU3J22-f4ahz3Erij63DZf7bSjh6uhIlCXtzx5Y,42835
@@ -451,10 +451,10 @@ diffusers/utils/dummy_torch_and_transformers_and_sentencepiece_objects.py,sha256
 diffusers/utils/dummy_torch_and_transformers_objects.py,sha256=SSjaTUruWwsbzR6pLv_5OU58YCGzq79gq7wW2Y9gSsY,61518
 diffusers/utils/dummy_transformers_and_torch_and_note_seq_objects.py,sha256=z-JrPgPo2dWv-buMytUqBd6QqEx8Uha6M1cKa6gR4Dc,621
 diffusers/utils/dynamic_modules_utils.py,sha256=WIsja-XIYgTXQ9-KK85uEU0l3rd_6yelUQCmb4nt0wU,19601
-diffusers/utils/export_utils.py,sha256=sPI6V4sX_2XZ3xXR1OBPAIiPS835NVMhlLHpuosxDKs,4412
+diffusers/utils/export_utils.py,sha256=2WSyPTf8oBmjx9cVI50AHe5LL72cvVYhPf_HpOIn_mU,6299
 diffusers/utils/hub_utils.py,sha256=EkQA7bmC6fCwAHuDhocjO5HEqmDDeWhYgf7VSEH_RTo,25966
-diffusers/utils/import_utils.py,sha256=b9IeqrbBk6yn3ALWnRQ4YHa2BMp9SVfTFFSGtym9hF4,29276
-diffusers/utils/loading_utils.py,sha256=8Nql6hZqYdVWMU_gbY-KYp6KVhIjQ0SbV9dz-MtxIfo,3867
+diffusers/utils/import_utils.py,sha256=BzTtGUO9giedRU_r2UmPtRpE2mqjvuxxd4BkwQQioV4,29942
+diffusers/utils/loading_utils.py,sha256=nEuLggTwPb_ixgLIaA12L0o2cj1fEs86yWE81H4xzOw,4025
 diffusers/utils/logging.py,sha256=9ghwwrmW92edOfS-FSFbRfVHO6EQXlDWPidfBag-JsI,9492
 diffusers/utils/model_card_template.md,sha256=ZhGhzjnMT2oPLbHnC0UYRNEpVC-okH-MLKjvkYsh-Ds,550
 diffusers/utils/outputs.py,sha256=hL1yMK86ota2SdZhyiWSzvfANjaO8pU7Hz1w81_Vmr8,4953
@@ -464,9 +464,9 @@ diffusers/utils/state_dict_utils.py,sha256=NsWzyX4eqKCfjLjgChQnFSf7nSQz1XFHgINYB
 diffusers/utils/testing_utils.py,sha256=16BNdTHQp9bJRkDi2iJa61i4ANC4875fjloTVmDIZ1w,36954
 diffusers/utils/torch_utils.py,sha256=6e6cJmPMbkEqXXJLlcKuVz0zZF8fIc7dF-azq-_6-Xw,6234
 diffusers/utils/versions.py,sha256=-e7XW1TzZ-tsRo9PMQHp-hNGYHuVDFzLtwg3uAJzqdI,4333
-diffusers-0.30.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-diffusers-0.30.0.dist-info/METADATA,sha256=kxFxUGTRQFO8TReJ-HCE7QtaJpEaMpQtpainaVpuJMI,18858
-diffusers-0.30.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-diffusers-0.30.0.dist-info/entry_points.txt,sha256=_1bvshKV_6_b63_FAkcUs9W6tUKGeIoQ3SHEZsovEWs,72
-diffusers-0.30.0.dist-info/top_level.txt,sha256=axJl2884vMSvhzrFrSoht36QXA_6gZN9cKtg4xOO72o,10
-diffusers-0.30.0.dist-info/RECORD,,
+diffusers-0.30.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+diffusers-0.30.2.dist-info/METADATA,sha256=5w7r16D1QCCi_8ZB1omljebUoUFZe7rLCC22aqXsItA,18858
+diffusers-0.30.2.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+diffusers-0.30.2.dist-info/entry_points.txt,sha256=_1bvshKV_6_b63_FAkcUs9W6tUKGeIoQ3SHEZsovEWs,72
+diffusers-0.30.2.dist-info/top_level.txt,sha256=axJl2884vMSvhzrFrSoht36QXA_6gZN9cKtg4xOO72o,10
+diffusers-0.30.2.dist-info/RECORD,,

{diffusers-0.30.0.dist-info → diffusers-0.30.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (74.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any