jbilcke-hf HF staff commited on
Commit
6c74560
1 Parent(s): 26effe4

Update handler.py

Browse files
Files changed (1) hide show
  1. 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=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",
 
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",