diff --git a/src/diffusers/pipelines/wan/pipeline_wan.py b/src/diffusers/pipelines/wan/pipeline_wan.py index 3294e9a56a07..814d6560b41b 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan.py +++ b/src/diffusers/pipelines/wan/pipeline_wan.py @@ -15,7 +15,6 @@ import html from typing import Any, Callable, Dict, List, Optional, Union -import ftfy import regex as re import torch from transformers import AutoTokenizer, UMT5EncoderModel @@ -24,7 +23,7 @@ from ...loaders import WanLoraLoaderMixin from ...models import AutoencoderKLWan, WanTransformer3DModel from ...schedulers import FlowMatchEulerDiscreteScheduler -from ...utils import is_torch_xla_available, logging, replace_example_docstring +from ...utils import is_ftfy_available, is_torch_xla_available, logging, replace_example_docstring from ...utils.torch_utils import randn_tensor from ...video_processor import VideoProcessor from ..pipeline_utils import DiffusionPipeline @@ -40,6 +39,9 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name +if is_ftfy_available(): + import ftfy + EXAMPLE_DOC_STRING = """ Examples: diff --git a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py index 775d86dcfc54..20ad84cb90d0 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_i2v.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_i2v.py @@ -15,7 +15,6 @@ import html from typing import Any, Callable, Dict, List, Optional, Tuple, Union -import ftfy import PIL import regex as re import torch @@ -26,7 +25,7 @@ from ...loaders import WanLoraLoaderMixin from ...models import AutoencoderKLWan, WanTransformer3DModel from ...schedulers import FlowMatchEulerDiscreteScheduler -from ...utils import is_torch_xla_available, logging, replace_example_docstring +from ...utils import is_ftfy_available, is_torch_xla_available, logging, replace_example_docstring from ...utils.torch_utils import randn_tensor from ...video_processor import VideoProcessor from ..pipeline_utils import DiffusionPipeline @@ -42,6 +41,9 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name +if is_ftfy_available(): + import ftfy + EXAMPLE_DOC_STRING = """ Examples: ```python diff --git a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py index c72dd7f5f1eb..e29267debbbb 100644 --- a/src/diffusers/pipelines/wan/pipeline_wan_video2video.py +++ b/src/diffusers/pipelines/wan/pipeline_wan_video2video.py @@ -16,7 +16,6 @@ import inspect from typing import Any, Callable, Dict, List, Optional, Union -import ftfy import regex as re import torch from PIL import Image @@ -26,7 +25,7 @@ from ...loaders import WanLoraLoaderMixin from ...models import AutoencoderKLWan, WanTransformer3DModel from ...schedulers import FlowMatchEulerDiscreteScheduler -from ...utils import is_torch_xla_available, logging, replace_example_docstring +from ...utils import is_ftfy_available, is_torch_xla_available, logging, replace_example_docstring from ...utils.torch_utils import randn_tensor from ...video_processor import VideoProcessor from ..pipeline_utils import DiffusionPipeline @@ -42,6 +41,9 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name +if is_ftfy_available(): + import ftfy + EXAMPLE_DOC_STRING = """ Examples: