List of pinned requirements / Dockerfile?
filipg7777 opened this issue · 2 comments
filipg7777 commented
I'm struggling to replicate DoReMi traning (weird errors, probably due to some incompatibility between dependencies).
Is there a list of pinned requirements (in particular what should be the version of flash_attn?). or, preferably, a Dockerfile for easy reproduction.
filipg7777 commented
Sorry, didn't notice that flash_attn should be v2.0.0. Anyway, the main question remains.
sangmichaelxie commented
The requirements should be here: https://raw.githubusercontent.com/sangmichaelxie/doremi/main/setup.py