Spaces:
Runtime error
Runtime error
Commit
·
976ac7c
1
Parent(s):
a8b3fe4
try rebuild with 50 ddim steps
Browse files- example.py +15 -2
- inference_fatezero.py +1 -1
example.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
num_steps =
|
| 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,
|
| 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 |
-
"
|
| 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)
|