| 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": { |
no outgoing calls