EasyLM
EasyLM copied to clipboard
fix the option to support 'streaming=false'
The original code can not support "streaming=false", resulting in the generated flax version of open_llama_7b can not be used in SecretFlow-SPU : OSError: Unable to convert /xxx/projects/open_llama_7b/flax_model.msgpack to Flax deserializable object