Skip to content

Update llama_model.py#2

Open
flyingdown wants to merge 1 commit into
ProjectD-AI:mainfrom
flyingdown:main
Open

Update llama_model.py#2
flyingdown wants to merge 1 commit into
ProjectD-AI:mainfrom
flyingdown:main

Conversation

@flyingdown
Copy link
Copy Markdown

添加LlamaParallelTransformerLayerPipe到checkpointable_layers

添加LlamaParallelTransformerLayerPipe到checkpointable_layers
@flyingdown
Copy link
Copy Markdown
Author

flyingdown commented Jul 20, 2023

可以避免反向找不到grad_fn的bug
RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant