adds llama and mistral dropout support (#858)

adds llama and mistral dropout support (#858) #2281