again
Browse files
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(
|