(model_manager: ModelManager, torch_dtype=None, device=None)
| 136 | |
| 137 | @staticmethod |
| 138 | def from_model_manager(model_manager: ModelManager, torch_dtype=None, device=None): |
| 139 | if device is None: device = model_manager.device |
| 140 | if torch_dtype is None: torch_dtype = model_manager.torch_dtype |
| 141 | pipe = FixPipeline(device=device, torch_dtype=torch_dtype) |
| 142 | pipe.fetch_models(model_manager) |
| 143 | return pipe |
| 144 | |
| 145 | |
| 146 | def denoising_model(self): |