Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	- modules.py +1 -1
    	
        modules.py
    CHANGED
    
    | @@ -2,7 +2,7 @@ import torch | |
| 2 | 
             
            import torch.nn as nn
         | 
| 3 | 
             
            from functools import partial
         | 
| 4 |  | 
| 5 | 
            -
            from  | 
| 6 |  | 
| 7 |  | 
| 8 | 
             
            class AbstractEncoder(nn.Module):
         | 
|  | |
| 2 | 
             
            import torch.nn as nn
         | 
| 3 | 
             
            from functools import partial
         | 
| 4 |  | 
| 5 | 
            +
            from x_transformer import Encoder, TransformerWrapper  # TODO: can we directly rely on lucidrains code and simply add this as a reuirement? --> test
         | 
| 6 |  | 
| 7 |  | 
| 8 | 
             
            class AbstractEncoder(nn.Module):
         | 
