graphstorm
graphstorm copied to clipboard
Modify the node classification module of peft_llm example
In https://github.com/awslabs/graphstorm/pull/760, we use GPEFT module as the gnn_encoder for link prediction. We still need to modify the node classification code to use GPEFT component and correct its model save function.