Method to

fastchat/model/rwkv_model.py:27–28

(self, target)

        self.model_path = model_path

    def to(self, target):
        assert target == "cuda"

    def __call__(self, input_ids, use_cache, past_key_values=None):
        assert use_cache == True