mayuema commited on
Commit
83c7862
1 Parent(s): defd900

first release

Browse files
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='/home/user/app/FollowYourPose/FollowYourPose/configs/pose_sample.yaml'
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()