adefossez commited on
Commit
c9179f8
1 Parent(s): 5c415d9
Files changed (1) hide show
  1. tests/modules/test_transformer.py +1 -1
tests/modules/test_transformer.py CHANGED
@@ -86,8 +86,8 @@ def test_streaming_api():
86
 
87
 
88
  def test_memory_efficient():
89
- torch.manual_seed(1234)
90
  for backend in ['torch', 'xformers']:
 
91
  set_efficient_attention_backend(backend)
92
 
93
  tr = StreamingTransformer(
 
86
 
87
 
88
  def test_memory_efficient():
 
89
  for backend in ['torch', 'xformers']:
90
+ torch.manual_seed(1234)
91
  set_efficient_attention_backend(backend)
92
 
93
  tr = StreamingTransformer(