Create testllm.py

testllm.py

@@ -0,0 +1,13 @@
+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")