Update handler.py
Browse files- handler.py +1 -4
handler.py
CHANGED
@@ -305,7 +305,7 @@ class EndpointHandler:
|
|
305 |
np.random.seed(config.seed)
|
306 |
generator = torch.manual_seed(config.seed)
|
307 |
|
308 |
-
# Prepare generation parameters for the video model (we omit params that are destined to Varnish)
|
309 |
generation_kwargs = {
|
310 |
# general content settings
|
311 |
"prompt": config.prompt,
|
@@ -317,9 +317,6 @@ class EndpointHandler:
|
|
317 |
"num_frames": config.num_frames,
|
318 |
"guidance_scale": config.guidance_scale,
|
319 |
"num_inference_steps": config.num_inference_steps,
|
320 |
-
|
321 |
-
# reproducible generation settings
|
322 |
-
"seed": config.seed,
|
323 |
|
324 |
# constants
|
325 |
"output_type": "pt",
|
|
|
305 |
np.random.seed(config.seed)
|
306 |
generator = torch.manual_seed(config.seed)
|
307 |
|
308 |
+
# Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
|
309 |
generation_kwargs = {
|
310 |
# general content settings
|
311 |
"prompt": config.prompt,
|
|
|
317 |
"num_frames": config.num_frames,
|
318 |
"guidance_scale": config.guidance_scale,
|
319 |
"num_inference_steps": config.num_inference_steps,
|
|
|
|
|
|
|
320 |
|
321 |
# constants
|
322 |
"output_type": "pt",
|