Rename && Update Readme
This commit is contained in:
+4
-4
@@ -15,9 +15,9 @@ from transformers import (CLIPImageProcessor, CLIPVisionModelWithProjection,
|
||||
|
||||
from cogvideox.models.autoencoder_magvit import AutoencoderKLCogVideoX
|
||||
from cogvideox.models.transformer3d import CogVideoXTransformer3DModel
|
||||
from cogvideox.pipeline.pipeline_cogvideox import CogVideoX_FUN_Pipeline
|
||||
from cogvideox.pipeline.pipeline_cogvideox import CogVideoX_Fun_Pipeline
|
||||
from cogvideox.pipeline.pipeline_cogvideox_inpaint import \
|
||||
CogVideoX_FUN_Pipeline_Inpaint
|
||||
CogVideoX_Fun_Pipeline_Inpaint
|
||||
from cogvideox.utils.lora_utils import merge_lora, unmerge_lora
|
||||
from cogvideox.utils.utils import get_video_to_video_latent, save_videos_grid
|
||||
|
||||
@@ -108,7 +108,7 @@ scheduler = Choosen_Scheduler.from_pretrained(
|
||||
)
|
||||
|
||||
if transformer.config.in_channels != vae.config.latent_channels:
|
||||
pipeline = CogVideoX_FUN_Pipeline_Inpaint.from_pretrained(
|
||||
pipeline = CogVideoX_Fun_Pipeline_Inpaint.from_pretrained(
|
||||
model_name,
|
||||
vae=vae,
|
||||
text_encoder=text_encoder,
|
||||
@@ -117,7 +117,7 @@ if transformer.config.in_channels != vae.config.latent_channels:
|
||||
torch_dtype=weight_dtype
|
||||
)
|
||||
else:
|
||||
pipeline = CogVideoX_FUN_Pipeline.from_pretrained(
|
||||
pipeline = CogVideoX_Fun_Pipeline.from_pretrained(
|
||||
model_name,
|
||||
vae=vae,
|
||||
text_encoder=text_encoder,
|
||||
|
||||
Reference in New Issue
Block a user