encounter1997 commited on
Commit
5d906f6
Β·
1 Parent(s): 248c435

use hf model

Browse files
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
- "./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,7 +94,7 @@ def create_demo(runner: Runner,
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,7 +104,7 @@ def create_demo(runner: Runner,
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,7 +114,7 @@ def create_demo(runner: Runner,
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,7 +124,7 @@ def create_demo(runner: Runner,
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,7 +134,7 @@ def create_demo(runner: Runner,
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.',
 
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
- # config.pretrained_model_path = self.download_base_model(model_path, token=input_token)
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