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

Class FakeModel

tests/test_llm_parser.py:682–689  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

680 return list(range(6))
681
682 class FakeModel:
683 def __init__(self, device):
684 self.device = device
685
686 def generate(self, **kwargs):
687 if self.device == "cuda":
688 raise RuntimeError("CUDA out of memory. Tried to allocate 38.00 MiB.")
689 return FakeOutputs()
690
691 backends = {
692 "cuda": {

Calls

no outgoing calls