testllm.py

from transformers import LlamaConfig, LlamaModel, LlamaForCausalLM, CLIPVisionModel, CLIPImageProcessor, \
    CLIPVisionConfig


class TestForCausalLM(LlamaForCausalLM):
    def __init__(self, config=None):
        super(LlamaForCausalLM, self).__init__(config)


    def forward(
            self,
    ):
        print("remote running success")