Maykeye
Maykeye
> yeah because you need to do os.environ['RWKV_CUDA_ON'] = '1' before import rwkv Yes, exactly! This is exactly is the issue. Normal imports look like this: ```python (BEGINNING OF FILE)...
> (i.e. user should fix it). The fix is that is appropriate for the user is changing `repeat(image, 'h -> (h H)', H=foo/bar) #incorrect call` to `repeat(image, 'h -> (h...
I had the same issue with pip. Workaround was to compile from source, worked as a charm ```python In [1]: import flash_attn In [2]: import torch In [3]: torch.__version__ Out[3]:...