MCPcopy Create free account
hub / github.com/OpenRaiser/PaperFlow / FakeTensor

Class FakeTensor

tests/test_llm_parser.py:652–663  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

650 monkeypatch.setattr(llm_parser, "LOCAL_LLM_DEVICE_OVERRIDE", None)
651
652 class FakeTensor:
653 def __init__(self, values):
654 self.values = list(values)
655 self.shape = (1, len(self.values))
656
657 def to(self, device):
658 return self
659
660 def __getitem__(self, item):
661 if isinstance(item, slice):
662 return self.values[item]
663 return self.values[item]
664
665 class FakeTokenizer:
666 pad_token_id = 0

Callers 1

__call__Method · 0.85

Calls

no outgoing calls

Tested by 1

__call__Method · 0.68