MusicGenerator

Runtime error

adefossez commited on Jun 13, 2023

Commit

c9179f8

•

1 Parent(s): 5c415d9

again

Files changed (1) hide show

tests/modules/test_transformer.py CHANGED Viewed

@@ -86,8 +86,8 @@ def test_streaming_api():
 def test_memory_efficient():
-    torch.manual_seed(1234)
     for backend in ['torch', 'xformers']:
         set_efficient_attention_backend(backend)
         tr = StreamingTransformer(

 def test_memory_efficient():
     for backend in ['torch', 'xformers']:
+        torch.manual_seed(1234)
         set_efficient_attention_backend(backend)
         tr = StreamingTransformer(