adefossez commited on
Commit
5c415d9
1 Parent(s): 86f6348

trying to fix test on CI

Browse files
Files changed (1) hide show
  1. tests/modules/test_transformer.py +1 -1
tests/modules/test_transformer.py CHANGED
@@ -132,8 +132,8 @@ def test_attention_as_float32():
132
 
133
  @torch.no_grad()
134
  def test_streaming_memory_efficient():
135
- torch.manual_seed(1234)
136
  for backend in ['torch', 'xformers']:
 
137
  set_efficient_attention_backend(backend)
138
  tr = StreamingTransformer(16, 4, 2, causal=True, dropout=0., custom=True)
139
  tr_mem_efficient = StreamingTransformer(
 
132
 
133
  @torch.no_grad()
134
  def test_streaming_memory_efficient():
 
135
  for backend in ['torch', 'xformers']:
136
+ torch.manual_seed(1234)
137
  set_efficient_attention_backend(backend)
138
  tr = StreamingTransformer(16, 4, 2, causal=True, dropout=0., custom=True)
139
  tr_mem_efficient = StreamingTransformer(