chenyangqi commited on
Commit
976ac7c
1 Parent(s): a8b3fe4

try rebuild with 50 ddim steps

Browse files
Files changed (2) hide show
  1. example.py +15 -2
  2. inference_fatezero.py +1 -1
example.py CHANGED
@@ -1,4 +1,4 @@
1
- num_steps = 10
2
  style_example = [
3
  [
4
  'CompVis/stable-diffusion-v1-4',
@@ -52,7 +52,7 @@ style_example = [
52
  10,
53
  num_steps,
54
  7.5,
55
- None, 0, 8, 1, 0,0,0,0
56
  ],
57
 
58
  [
@@ -68,5 +68,18 @@ style_example = [
68
  7.5,
69
  None, 0, 8, 1, 0,0,0,0
70
  ],
 
 
 
 
 
 
 
 
 
 
 
 
 
71
 
72
  ]
1
+ num_steps = 50
2
  style_example = [
3
  [
4
  'CompVis/stable-diffusion-v1-4',
52
  10,
53
  num_steps,
54
  7.5,
55
+ None, 0, 8, 28, 0,0,0,0
56
  ],
57
 
58
  [
68
  7.5,
69
  None, 0, 8, 1, 0,0,0,0
70
  ],
71
+ [
72
+ 'CompVis/stable-diffusion-v1-4',
73
+ 'FateZero/data/attribute/squirrel_carrot.mp4',
74
+ 'A squirrel is eating a carrot',
75
+ 'A rabbit is eating a eggplant',
76
+ 0.5,
77
+ 0.5,
78
+ 'rabbit eggplant',
79
+ 10,
80
+ num_steps,
81
+ 7.5,
82
+ None, 0, 8, 1, 0,0,0,0
83
+ ],
84
 
85
  ]
inference_fatezero.py CHANGED
@@ -86,7 +86,7 @@ class merge_config_then_run():
86
  ImageSequenceDataset_dict = {
87
  "start_sample_frame" : start_sample_frame,
88
  "n_sample_frame" : n_sample_frame,
89
- "stride" : stride,
90
  "offset": offset_dict,
91
  }
92
  config_now['train_dataset'].update(ImageSequenceDataset_dict)
86
  ImageSequenceDataset_dict = {
87
  "start_sample_frame" : start_sample_frame,
88
  "n_sample_frame" : n_sample_frame,
89
+ "sampling_rate" : stride,
90
  "offset": offset_dict,
91
  }
92
  config_now['train_dataset'].update(ImageSequenceDataset_dict)