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.
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.