AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

update

Browse Source
This commit is contained in:
Dhruv Nair
2024-10-24 22:50:54 +02:00
parent 8700d64d62
commit 969c3aba88
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/pipelines/mochi/pipeline_mochi.py
View File

@@ -18,7 +18,6 @@ from typing import Any, Callable, Dict, List, Optional, Union
import torch
from transformers import T5EncoderModel, T5TokenizerFast
from ...image_processor import VaeImageProcessor
from ...loaders import TextualInversionLoaderMixin
from ...models.autoencoders import AutoencoderKL
from ...models.transformers import MochiTransformer3DModel
@@ -29,6 +28,7 @@ from ...utils import (
replace_example_docstring,
)
from ...utils.torch_utils import randn_tensor
from ...video_processor import VideoProcessor
from ..pipeline_utils import DiffusionPipeline
from .pipeline_output import MochiPipelineOutput
@@ -205,7 +205,7 @@ class MochiPipeline(DiffusionPipeline, TextualInversionLoaderMixin):
self.vae_temporal_scale_factor = 6
self.patch_size = 2
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
self.video_processor = VideoProcessor(vae_scale_factor=self.vae_scale_factor)
self.tokenizer_max_length = (
self.tokenizer.model_max_length if hasattr(self, "tokenizer") and self.tokenizer is not None else 77
)