File size: 347 Bytes
b2dfe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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")
|