neeraj1bh commited on
Commit
199858f
1 Parent(s): e66f43c

chore: added config

Browse files
Files changed (1) hide show
  1. config.json +49 -0
config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_id": "MagicAnimate",
3
+ "description": "Temporally Consistent Human Image Animation using Diffusion Model",
4
+ "license": "BSD-3-Clause",
5
+ "version": "1.0",
6
+ "configurations": [
7
+ {
8
+ "model": "pretrained_models/stable-diffusion-v1-5",
9
+ "description": "Stable Diffusion v1.5 model with various enhancements for temporal consistency.",
10
+ "config_file": "pretrained_models/stable-diffusion-v1-5/config.json"
11
+ },
12
+ {
13
+ "model": "pretrained_models/sd-vae-ft-mse",
14
+ "description": "VAE finetuned with MSE loss for better latent representations.",
15
+ "config_file": "pretrained_models/sd-vae-ft-mse/config.json"
16
+ },
17
+ {
18
+ "model": "pretrained_models/MagicAnimate/appearance_encoder",
19
+ "description": "Appearance encoder for extracting and encoding appearance features.",
20
+ "config_file": "pretrained_models/MagicAnimate/appearance_encoder/config.json"
21
+ },
22
+ {
23
+ "model": "pretrained_models/MagicAnimate/densepose_controlnet",
24
+ "description": "DensePose control net for controlling animations based on dense pose inputs.",
25
+ "config_file": "pretrained_models/MagicAnimate/densepose_controlnet/config.json"
26
+ },
27
+ {
28
+ "model": "pretrained_models/MagicAnimate/temporal_attention",
29
+ "description": "Temporal attention model for ensuring temporal consistency in animations.",
30
+ "config_file": "pretrained_models/MagicAnimate/temporal_attention/config.json"
31
+ }
32
+ ],
33
+ "dependencies": {
34
+ "python_version": "3.8",
35
+ "packages": [
36
+ "torch==2.0.1",
37
+ "torchvision==0.15.2",
38
+ "diffusers==0.21.4",
39
+ "transformers==4.32.0",
40
+ "fastapi==0.103.0",
41
+ "gradio==3.41.2",
42
+ "aiohttp==3.8.5",
43
+ "uvicorn==0.23.2",
44
+ "numpy==1.24.4",
45
+ "opencv-python==4.8.0.76",
46
+ "scipy==1.10.1"
47
+ ]
48
+ }
49
+ }