Spaces:
Runtime error
Runtime error
mayuema
commited on
Commit
•
83c7862
1
Parent(s):
defd900
first release
Browse files- FollowYourPose/.gitignore +0 -51
- FollowYourPose/__pycache__/test_followyourpose.cpython-38.pyc +0 -0
- FollowYourPose/checkpoints +1 -0
- FollowYourPose/configs/pose_sample.yaml +28 -0
- FollowYourPose/followyourpose/__pycache__/util.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/data/__pycache__/dataset.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/data/__pycache__/hdvila.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/models/__pycache__/attention.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/models/__pycache__/resnet.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/models/__pycache__/unet.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/models/__pycache__/unet_blocks.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/pipelines/__pycache__/pipeline_followyourpose.cpython-38.pyc +0 -0
- FollowYourPose/followyourpose/pipelines/__pycache__/pipeline_tuneavideo.cpython-38.pyc +0 -0
- __pycache__/example.cpython-38.pyc +0 -0
- __pycache__/inference_followyourpose.cpython-38.pyc +0 -0
- app.py +1 -0
- inference_followyourpose.py +2 -2
FollowYourPose/.gitignore
DELETED
@@ -1,51 +0,0 @@
|
|
1 |
-
|
2 |
-
/checkpoints
|
3 |
-
|
4 |
-
/data
|
5 |
-
|
6 |
-
/sim_matrix
|
7 |
-
|
8 |
-
/figs
|
9 |
-
|
10 |
-
/log
|
11 |
-
|
12 |
-
|
13 |
-
*.log
|
14 |
-
|
15 |
-
*.sh
|
16 |
-
|
17 |
-
*.pth
|
18 |
-
|
19 |
-
*.jpg
|
20 |
-
|
21 |
-
*.og
|
22 |
-
|
23 |
-
/logs
|
24 |
-
|
25 |
-
/stable_diffusion.egg-info
|
26 |
-
|
27 |
-
/__pycache__
|
28 |
-
|
29 |
-
/outputs
|
30 |
-
|
31 |
-
*.pyc
|
32 |
-
|
33 |
-
/models
|
34 |
-
|
35 |
-
*.ckpt
|
36 |
-
|
37 |
-
/src
|
38 |
-
|
39 |
-
/newsd_weight
|
40 |
-
|
41 |
-
*.yaml
|
42 |
-
|
43 |
-
*.png
|
44 |
-
|
45 |
-
*.yaml
|
46 |
-
|
47 |
-
*.txt
|
48 |
-
|
49 |
-
/data
|
50 |
-
|
51 |
-
/others
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
FollowYourPose/__pycache__/test_followyourpose.cpython-38.pyc
ADDED
Binary file (4.99 kB). View file
|
|
FollowYourPose/checkpoints
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
/apdcephfs/private_mayuema/FollowYourPose/FollowYourPose/checkpoints
|
FollowYourPose/configs/pose_sample.yaml
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
pretrained_model_path: "./FollowYourPose/checkpoints/stable-diffusion-v1-4"
|
2 |
+
output_dir: "output"
|
3 |
+
|
4 |
+
|
5 |
+
validation_data:
|
6 |
+
prompts:
|
7 |
+
- "Stormtrooper on the sea"
|
8 |
+
- "Astronaut on the beach"
|
9 |
+
video_length: 32
|
10 |
+
width: 512
|
11 |
+
height: 512
|
12 |
+
num_inference_steps: 50
|
13 |
+
guidance_scale: 12.5
|
14 |
+
use_inv_latent: False
|
15 |
+
num_inv_steps: 50
|
16 |
+
dataset_set: "val"
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
train_batch_size: 1
|
21 |
+
validation_steps: 100
|
22 |
+
skeleton_path: ' '
|
23 |
+
resume_from_checkpoint: './FollowYourPose/checkpoints/followyourpose_checkpoint-1000'
|
24 |
+
|
25 |
+
seed: 33
|
26 |
+
mixed_precision: 'no'
|
27 |
+
gradient_checkpointing: False
|
28 |
+
enable_xformers_memory_efficient_attention: True
|
FollowYourPose/followyourpose/__pycache__/util.cpython-38.pyc
ADDED
Binary file (2.92 kB). View file
|
|
FollowYourPose/followyourpose/data/__pycache__/dataset.cpython-38.pyc
ADDED
Binary file (1.5 kB). View file
|
|
FollowYourPose/followyourpose/data/__pycache__/hdvila.cpython-38.pyc
ADDED
Binary file (177 Bytes). View file
|
|
FollowYourPose/followyourpose/models/__pycache__/attention.cpython-38.pyc
ADDED
Binary file (8.93 kB). View file
|
|
FollowYourPose/followyourpose/models/__pycache__/resnet.cpython-38.pyc
ADDED
Binary file (5.12 kB). View file
|
|
FollowYourPose/followyourpose/models/__pycache__/unet.cpython-38.pyc
ADDED
Binary file (14.9 kB). View file
|
|
FollowYourPose/followyourpose/models/__pycache__/unet_blocks.cpython-38.pyc
ADDED
Binary file (11.6 kB). View file
|
|
FollowYourPose/followyourpose/pipelines/__pycache__/pipeline_followyourpose.cpython-38.pyc
ADDED
Binary file (12.7 kB). View file
|
|
FollowYourPose/followyourpose/pipelines/__pycache__/pipeline_tuneavideo.cpython-38.pyc
ADDED
Binary file (12.6 kB). View file
|
|
__pycache__/example.cpython-38.pyc
ADDED
Binary file (521 Bytes). View file
|
|
__pycache__/inference_followyourpose.cpython-38.pyc
CHANGED
Binary files a/__pycache__/inference_followyourpose.cpython-38.pyc and b/__pycache__/inference_followyourpose.cpython-38.pyc differ
|
|
app.py
CHANGED
@@ -171,3 +171,4 @@ with gr.Blocks(css='style.css') as demo:
|
|
171 |
run_button.click(fn=pipe.run, inputs=inputs, outputs=result)
|
172 |
|
173 |
demo.queue().launch()
|
|
|
|
171 |
run_button.click(fn=pipe.run, inputs=inputs, outputs=result)
|
172 |
|
173 |
demo.queue().launch()
|
174 |
+
# demo.queue().launch(share=False, server_name='0.0.0.0', server_port=7890)
|
inference_followyourpose.py
CHANGED
@@ -20,7 +20,7 @@ class merge_config_then_run():
|
|
20 |
self.text_encoder = None
|
21 |
self.vae = None
|
22 |
self.unet = None
|
23 |
-
self.download_model()
|
24 |
|
25 |
def download_model(self):
|
26 |
REPO_ID = 'YueMafighting/FollowYourPose_v1'
|
@@ -43,7 +43,7 @@ class merge_config_then_run():
|
|
43 |
bottom_crop=0,
|
44 |
):
|
45 |
|
46 |
-
default_edit_config='
|
47 |
Omegadict_default_edit_config = OmegaConf.load(default_edit_config)
|
48 |
|
49 |
dataset_time_string = get_time_string()
|
|
|
20 |
self.text_encoder = None
|
21 |
self.vae = None
|
22 |
self.unet = None
|
23 |
+
# self.download_model()
|
24 |
|
25 |
def download_model(self):
|
26 |
REPO_ID = 'YueMafighting/FollowYourPose_v1'
|
|
|
43 |
bottom_crop=0,
|
44 |
):
|
45 |
|
46 |
+
default_edit_config='./FollowYourPose/configs/pose_sample.yaml'
|
47 |
Omegadict_default_edit_config = OmegaConf.load(default_edit_config)
|
48 |
|
49 |
dataset_time_string = get_time_string()
|