Spaces:
Runtime error
Runtime error
encounter1997
commited on
Commit
Β·
5d906f6
1
Parent(s):
248c435
use hf model
Browse files- gradio_demo/app_running.py +6 -6
- gradio_demo/runner.py +1 -2
gradio_demo/app_running.py
CHANGED
@@ -84,7 +84,7 @@ def create_demo(runner: Runner,
|
|
84 |
examples = [
|
85 |
[
|
86 |
'CompVis/stable-diffusion-v1-4',
|
87 |
-
"
|
88 |
'A car is moving on the road.',
|
89 |
8, 0, 1,
|
90 |
'A jeep car is moving on the desert.',
|
@@ -94,7 +94,7 @@ def create_demo(runner: Runner,
|
|
94 |
|
95 |
[
|
96 |
'CompVis/stable-diffusion-v1-4',
|
97 |
-
"
|
98 |
'A blackswan is swimming on the water.',
|
99 |
8, 0, 4,
|
100 |
'A white swan is swimming on the water.',
|
@@ -104,7 +104,7 @@ def create_demo(runner: Runner,
|
|
104 |
|
105 |
[
|
106 |
'CompVis/stable-diffusion-v1-4',
|
107 |
-
"
|
108 |
'A child is riding a bike on the road.',
|
109 |
8, 0, 1,
|
110 |
'A lego child is riding a bike on the road.',
|
@@ -114,7 +114,7 @@ def create_demo(runner: Runner,
|
|
114 |
|
115 |
[
|
116 |
'CompVis/stable-diffusion-v1-4',
|
117 |
-
"
|
118 |
'A jeep car is moving on the road.',
|
119 |
8, 0, 6,
|
120 |
'A jeep car is moving on the snow.',
|
@@ -124,7 +124,7 @@ def create_demo(runner: Runner,
|
|
124 |
|
125 |
[
|
126 |
'CompVis/stable-diffusion-v1-4',
|
127 |
-
"
|
128 |
'A rabbit is eating a watermelon.',
|
129 |
8, 0, 6,
|
130 |
'A puppy is eating an orange.',
|
@@ -134,7 +134,7 @@ def create_demo(runner: Runner,
|
|
134 |
|
135 |
[
|
136 |
'CompVis/stable-diffusion-v1-4',
|
137 |
-
"
|
138 |
'A brown bear is sitting on the ground.',
|
139 |
8, 0, 6,
|
140 |
'A black bear is sitting on the grass.',
|
|
|
84 |
examples = [
|
85 |
[
|
86 |
'CompVis/stable-diffusion-v1-4',
|
87 |
+
"data/car-moving.mp4",
|
88 |
'A car is moving on the road.',
|
89 |
8, 0, 1,
|
90 |
'A jeep car is moving on the desert.',
|
|
|
94 |
|
95 |
[
|
96 |
'CompVis/stable-diffusion-v1-4',
|
97 |
+
"data/black-swan.mp4",
|
98 |
'A blackswan is swimming on the water.',
|
99 |
8, 0, 4,
|
100 |
'A white swan is swimming on the water.',
|
|
|
104 |
|
105 |
[
|
106 |
'CompVis/stable-diffusion-v1-4',
|
107 |
+
"data/child-riding.mp4",
|
108 |
'A child is riding a bike on the road.',
|
109 |
8, 0, 1,
|
110 |
'A lego child is riding a bike on the road.',
|
|
|
114 |
|
115 |
[
|
116 |
'CompVis/stable-diffusion-v1-4',
|
117 |
+
"data/car-turn.mp4",
|
118 |
'A jeep car is moving on the road.',
|
119 |
8, 0, 6,
|
120 |
'A jeep car is moving on the snow.',
|
|
|
124 |
|
125 |
[
|
126 |
'CompVis/stable-diffusion-v1-4',
|
127 |
+
"data/rabbit-watermelon.mp4",
|
128 |
'A rabbit is eating a watermelon.',
|
129 |
8, 0, 6,
|
130 |
'A puppy is eating an orange.',
|
|
|
134 |
|
135 |
[
|
136 |
'CompVis/stable-diffusion-v1-4',
|
137 |
+
"data/brown-bear.mp4",
|
138 |
'A brown bear is sitting on the ground.',
|
139 |
8, 0, 6,
|
140 |
'A black bear is sitting on the grass.',
|
gradio_demo/runner.py
CHANGED
@@ -88,8 +88,7 @@ class Runner:
|
|
88 |
output_dir.mkdir(parents=True, exist_ok=True)
|
89 |
|
90 |
config = OmegaConf.load('configs/black-swan.yaml')
|
91 |
-
|
92 |
-
config.pretrained_model_path = "checkpoints/stable-diffusion-v1-4" # TODO
|
93 |
|
94 |
config.output_dir = output_dir.as_posix()
|
95 |
config.input_data.video_path = input_video.name # type: ignore
|
|
|
88 |
output_dir.mkdir(parents=True, exist_ok=True)
|
89 |
|
90 |
config = OmegaConf.load('configs/black-swan.yaml')
|
91 |
+
config.pretrained_model_path = self.download_base_model(model_path, token=input_token)
|
|
|
92 |
|
93 |
config.output_dir = output_dir.as_posix()
|
94 |
config.input_data.video_path = input_video.name # type: ignore
|