Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
@@ -405,8 +405,9 @@ class EndpointHandler:
|
|
405 |
# Set random seeds
|
406 |
random.seed(config.seed)
|
407 |
np.random.seed(config.seed)
|
408 |
-
generator = torch.Generator(device='cuda')
|
409 |
torch.manual_seed(config.seed)
|
|
|
|
|
410 |
|
411 |
# Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
|
412 |
generation_kwargs = {
|
|
|
405 |
# Set random seeds
|
406 |
random.seed(config.seed)
|
407 |
np.random.seed(config.seed)
|
|
|
408 |
torch.manual_seed(config.seed)
|
409 |
+
generator = torch.Generator(device='cuda')
|
410 |
+
generator = generator.manual_seed(config.seed)
|
411 |
|
412 |
# Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
|
413 |
generation_kwargs = {
|