From 202ae9fdfe7f638e5bdd0b8cadc6ab83851b7edd Mon Sep 17 00:00:00 2001 From: LiewFeng <1361871897@qq.com> Date: Fri, 27 Dec 2024 10:55:22 +0800 Subject: [PATCH] fix typo --- README.md | 2 +- eval/teacache/experiments/cogvideox.py | 4 ++-- eval/teacache/experiments/latte.py | 7 +++++-- eval/teacache/experiments/opensora.py | 4 ++-- eval/teacache/experiments/opensora_plan.py | 4 ++-- 5 files changed, 12 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index e6c9b71..d55fe1a 100644 --- a/README.md +++ b/README.md @@ -85,7 +85,7 @@ conda create -n teacache python=3.10 -y conda activate teacache ``` -Install VideoSys: +Install TeaCache: ```shell git clone https://github.com/LiewFeng/TeaCache diff --git a/eval/teacache/experiments/cogvideox.py b/eval/teacache/experiments/cogvideox.py index 236e00a..e2a4ac1 100644 --- a/eval/teacache/experiments/cogvideox.py +++ b/eval/teacache/experiments/cogvideox.py @@ -200,7 +200,7 @@ def eval_teacache_slow(prompt_list): engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None engine.driver_worker.transformer.__class__.previous_residual_encoder = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/cogvideox_teacache_slow", loop=5) def eval_teacache_fast(prompt_list): @@ -212,7 +212,7 @@ def eval_teacache_fast(prompt_list): engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None engine.driver_worker.transformer.__class__.previous_residual_encoder = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/cogvideox_teacache_fast", loop=5) diff --git a/eval/teacache/experiments/latte.py b/eval/teacache/experiments/latte.py index 167ecb8..107078e 100644 --- a/eval/teacache/experiments/latte.py +++ b/eval/teacache/experiments/latte.py @@ -6,6 +6,9 @@ from torch import nn import numpy as np from typing import Any, Dict, Optional, Tuple from videosys.core.comm import all_to_all_with_pad, gather_sequence, get_pad, set_pad, split_sequence +from videosys.models.transformers.latte_transformer_3d import Transformer3DModelOutput +from videosys.utils.utils import batch_func +from functools import partial def teacache_forward( self, @@ -502,7 +505,7 @@ def eval_teacache_slow(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/latte_teacache_slow", loop=5) def eval_teacache_fast(prompt_list): @@ -513,7 +516,7 @@ def eval_teacache_fast(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/latte_teacache_fast", loop=5) diff --git a/eval/teacache/experiments/opensora.py b/eval/teacache/experiments/opensora.py index 561371d..eb1966e 100644 --- a/eval/teacache/experiments/opensora.py +++ b/eval/teacache/experiments/opensora.py @@ -216,7 +216,7 @@ def eval_teacache_slow(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/opensora_teacache_slow", loop=5) def eval_teacache_fast(prompt_list): @@ -227,7 +227,7 @@ def eval_teacache_fast(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/opensora_teacache_fast", loop=5) diff --git a/eval/teacache/experiments/opensora_plan.py b/eval/teacache/experiments/opensora_plan.py index 408c267..671e58c 100644 --- a/eval/teacache/experiments/opensora_plan.py +++ b/eval/teacache/experiments/opensora_plan.py @@ -565,7 +565,7 @@ def eval_teacache_slow(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/opensoraplan_teacache_slow", loop=5) def eval_teacache_fast(prompt_list): @@ -576,7 +576,7 @@ def eval_teacache_fast(prompt_list): engine.driver_worker.transformer.__class__.accumulated_rel_l1_distance = 0 engine.driver_worker.transformer.__class__.previous_modulated_input = None engine.driver_worker.transformer.__class__.previous_residual = None - engine.driver_worker.transformer.__class__.__class__.forward = teacache_forward + engine.driver_worker.transformer.__class__.forward = teacache_forward generate_func(engine, prompt_list, "./samples/opensoraplan_teacache_fast", loop=5)