Update README.md

README.md

@@ -7,6 +7,81 @@ license: openrail++
 inference: false
 ---
 
+# What is different about this fork from the original (h1t/oms_b_openclip_xl)?
+
+The code has been modified to work with the current final version (0.27.2) of diffusers.  
+The behavior remains the same. Enjoy.
+
+<img src="kaeru-dev.png" width="600"/>
+
+```diff
+- OMSPipeline.from_pretrained('h1t/oms_b_openclip_xl', ...)
++ OMSPipeline.from_pretrained('kaeru-shigure/oms_b_openclip_xl', ...)
+```
+
+```diff
+--- a/diffusers_patch/models/unet_2d_condition_woct.py
++++ b/diffusers_patch/models/unet_2d_condition_woct.py
+@@ -35,7 +35,7 @@ from diffusers.models.embeddings import (
+     Timesteps,
+ )
+ from diffusers.models.modeling_utils import ModelMixin
+-from diffusers.models.unet_2d_blocks import (
++from diffusers.models.unets.unet_2d_blocks import (
+     CrossAttnDownBlock2D,
+     CrossAttnUpBlock2D,
+     DownBlock2D,
+@@ -159,6 +159,7 @@ class UNet2DConditionWoCTModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMi
+         conv_out_kernel: int = 3,
+         mid_block_only_cross_attention: Optional[bool] = None,
+         cross_attention_norm: Optional[str] = None,
++        subfolder: Optional[str] = None,
+     ):
+         super().__init__()
+```
+
+```diff
+--- a/diffusers_patch/pipelines/oms/pipeline_oms.py
++++ b/diffusers_patch/pipelines/oms/pipeline_oms.py
+@@ -8,6 +8,7 @@ from transformers import CLIPTextModel, CLIPTextModelWithProjection, CLIPTokeniz
+
+ from diffusers.loaders import FromSingleFileMixin
+
++from huggingface_hub.constants import HF_HUB_CACHE, HF_HUB_OFFLINE
+ from diffusers.utils import (
+     USE_PEFT_BACKEND,
+     deprecate,
+@@ -17,6 +18,7 @@ from diffusers.utils.torch_utils import randn_tensor
+ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
+ from diffusers.pipelines.pipeline_utils import *
+ from diffusers.pipelines.pipeline_utils import _get_pipeline_class
++from diffusers.pipelines.pipeline_loading_utils import *
+ from diffusers.models.modeling_utils import _LOW_CPU_MEM_USAGE_DEFAULT
+
+ from diffusers_patch.models.unet_2d_condition_woct import UNet2DConditionWoCTModel
+@@ -164,7 +166,8 @@ class OMSPipeline(DiffusionPipeline, FromSingleFileMixin):
+         sd_pipeline: DiffusionPipeline,
+         oms_text_encoder:Optional[Union[CLIPTextModel, SDXLTextEncoder]],
+         oms_tokenizer:Optional[Union[CLIPTokenizer, SDXLTokenizer]],
+-        sd_scheduler = None
++        sd_scheduler = None,
++        trust_remote_code: bool = False,
+     ):
+         # assert sd_pipeline is not None
+
+@@ -279,7 +282,7 @@ class OMSPipeline(DiffusionPipeline, FromSingleFileMixin):
+
+     @classmethod
+ os.PathLike]], **kwargs):
+-        cache_dir = kwargs.pop("cache_dir", DIFFUSERS_CACHE)
++        cache_dir = kwargs.pop("cache_dir", HF_HUB_CACHE)
+         resume_download = kwargs.pop("resume_download", False)
+         force_download = kwargs.pop("force_download", False)
+         proxies = kwargs.pop("proxies", None)
+```
+
+-----
+
 # One More Step
 
 One More Step (OMS) module was proposed in [One More Step: A Versatile Plug-and-Play Module for Rectifying Diffusion Schedule Flaws and Enhancing Low-Frequency Controls](https://github.com/mhh0318/OneMoreStep)