(
self,
torch_dtype = torch.float16,
device = "cuda",
)
| 110 | |
| 111 | class ModelManager: |
| 112 | def __init__( |
| 113 | self, |
| 114 | torch_dtype = torch.float16, |
| 115 | device = "cuda", |
| 116 | ): |
| 117 | self.torch_dtype = torch_dtype |
| 118 | self.device = device |
| 119 | self.model = [] |
| 120 | self.model_path = [] |
| 121 | self.model_name = [] |
| 122 | |
| 123 | def load_model(self, file_path, model_type=None, device=None, torch_dtype=None): |
| 124 | print(f"Loading models from: {file_path}") |
nothing calls this directly
no outgoing calls
no test coverage detected