Can you provide a requirements.txt for installing?

OliverRensu / ARM

This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision

53 stars 1 forks source link

Can you provide a requirements.txt for installing? #2

Open showstarpro opened 2 months ago

showstarpro commented 2 months ago

To better reproduce this article of yours, could you list the pytorch version number of this repository? If you can export a direct requirements.txt, we would be very grateful.

OliverRensu commented 2 months ago

Hi please follow Vim https://github.com/hustvl/Vim and flashattention https://github.com/Dao-AILab/flash-attention