jbilcke-hf HF staff commited on
Commit
c6a43f8
·
verified ·
1 Parent(s): 0f751a9

Update handler.py

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