Mirrors/AnimateDiff
1
0
Fork 0
mirror of https://github.com/guoyww/AnimateDiff.git synced 2025-12-24 04:09:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

gradio

Browse Source
This commit is contained in:
Yuwei Guo
2023-07-18 17:03:31 +08:00
parent 07236ae2da
commit 170005366a
1 changed files with 8 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
app.py
View File

@@ -1,22 +1,18 @@
import gradio as gr
import os
from glob import glob
import random
import pdb
from transformers import CLIPTextModel, CLIPTokenizer
from animatediff.models.unet import UNet3DConditionModel
from animatediff.pipelines.pipeline_animation import AnimationPipeline
from diffusers import AutoencoderKL
from datetime import datetime
import os import os
from omegaconf import OmegaConf
import json import json
import torch import torch
import random
import gradio as gr
from glob import glob
from omegaconf import OmegaConf
from datetime import datetime
from safetensors import safe_open
from diffusers import AutoencoderKL from diffusers import AutoencoderKL
from diffusers import DDIMScheduler, EulerDiscreteScheduler, PNDMScheduler from diffusers import DDIMScheduler, EulerDiscreteScheduler, PNDMScheduler
from diffusers.utils.import_utils import is_xformers_available
from transformers import CLIPTextModel, CLIPTokenizer from transformers import CLIPTextModel, CLIPTokenizer
from animatediff.models.unet import UNet3DConditionModel from animatediff.models.unet import UNet3DConditionModel
@@ -24,13 +20,9 @@ from animatediff.pipelines.pipeline_animation import AnimationPipeline
from animatediff.utils.util import save_videos_grid from animatediff.utils.util import save_videos_grid
from animatediff.utils.convert_from_ckpt import convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint from animatediff.utils.convert_from_ckpt import convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint
from animatediff.utils.convert_lora_safetensor_to_diffusers import convert_lora from animatediff.utils.convert_lora_safetensor_to_diffusers import convert_lora
from diffusers.utils.import_utils import is_xformers_available
from safetensors import safe_open
sample_idx = 0 sample_idx = 0
scheduler_dict = { scheduler_dict = {
"Euler": EulerDiscreteScheduler, "Euler": EulerDiscreteScheduler,
"PNDM": PNDMScheduler, "PNDM": PNDMScheduler,
@@ -46,7 +38,6 @@ css = """
} }
""" """
class AnimateController: class AnimateController:
def __init__(self): def __init__(self):