yunyang GeandLinB203
842435c016
* gitignore * inpaint * Create condition_image_path.txt * Rename condition_image_path.txt to condition_images_path.txt * Update sample_inpaint.py * Update sample_inpaint.sh * inpaint * fix bug * Update README.md * inpainting * release v1.2.0 i2v * release v1.2.0 i2v * Update Report-v1.2.0.md * Update pipeline_inpaint_sp.py * Update sample_inpaint_ddp.py * Update sample_inpaint_sp.py * Update sample_inpaint.py * Grammar Error Correction * Grammar Error Correction --------- Co-authored-by: LinB203 <2267330597@qq.com>
21 lines
690 B
Python
21 lines
690 B
Python
|
|
from .opensora.modeling_opensora import OpenSora_models
|
|
from .udit.modeling_udit import UDiT_models
|
|
from .opensora.modeling_inpaint import OpenSoraInpaint_models
|
|
|
|
|
|
Diffusion_models = {}
|
|
Diffusion_models.update(OpenSora_models)
|
|
Diffusion_models.update(UDiT_models)
|
|
Diffusion_models.update(OpenSoraInpaint_models)
|
|
|
|
from .opensora.modeling_opensora import OpenSora_models_class
|
|
from .udit.modeling_udit import UDiT_models_class
|
|
from .opensora.modeling_inpaint import OpenSoraInpaint_models_class
|
|
|
|
|
|
Diffusion_models_class = {}
|
|
Diffusion_models_class.update(OpenSora_models_class)
|
|
Diffusion_models_class.update(UDiT_models_class)
|
|
Diffusion_models_class.update(OpenSoraInpaint_models_class)
|
|
|