Commit
•
6c74560
1
Parent(s):
26effe4
Update handler.py
Browse files- handler.py +8 -8
handler.py
CHANGED
@@ -290,18 +290,18 @@ class EndpointHandler:
|
|
290 |
# Prepare generation parameters for the video model (we omit params that are destined to Varnish)
|
291 |
generation_kwargs = {
|
292 |
# general content settings
|
293 |
-
prompt: config.prompt,
|
294 |
-
negative_prompt
|
295 |
|
296 |
# video model settings (will be used during generation of the initial raw video clip)
|
297 |
-
width
|
298 |
-
height
|
299 |
-
num_frames
|
300 |
-
guidance_scale
|
301 |
-
num_inference_steps
|
302 |
|
303 |
# reproducible generation settings
|
304 |
-
seed
|
305 |
|
306 |
# constants
|
307 |
"output_type": "pt",
|
|
|
290 |
# Prepare generation parameters for the video model (we omit params that are destined to Varnish)
|
291 |
generation_kwargs = {
|
292 |
# general content settings
|
293 |
+
"prompt": config.prompt,
|
294 |
+
"negative_prompt": config.negative_prompt,
|
295 |
|
296 |
# video model settings (will be used during generation of the initial raw video clip)
|
297 |
+
"width": params.config.width,
|
298 |
+
"height": config.height,
|
299 |
+
"num_frames": config.num_frames,
|
300 |
+
"guidance_scale": config.guidance_scale,
|
301 |
+
"num_inference_steps": config.num_inference_steps,
|
302 |
|
303 |
# reproducible generation settings
|
304 |
+
"seed": config.seed,
|
305 |
|
306 |
# constants
|
307 |
"output_type": "pt",
|