improve(pipeline): track all Module instances (not just Model's) (#1646)

CHANGELOG.md

@@ -21,6 +21,7 @@
 ### Improvements
 
 - improve(metric): add support for number of speakers mismatch in `diarization_error_rate` metric
+- improve(pipeline): track both `Model` and `nn.Module` attributes in `Pipeline.to(device)`
 
 ## Breaking changes
 

pyannote/audio/core/pipeline.py

@@ -29,6 +29,7 @@
 from typing import Callable, Dict, List, Optional, Text, Union
 
 import torch
+import torch.nn as nn
 import yaml
 from huggingface_hub import hf_hub_download
 from huggingface_hub.utils import RepositoryNotFoundError
@@ -232,7 +233,7 @@ def remove_from(*dicts):
         _models = self.__dict__.get("_models")
         _inferences = self.__dict__.get("_inferences")
 
-        if isinstance(value, Model):
+        if isinstance(value, nn.Module):
             if _models is None:
                 msg = "cannot assign models before Pipeline.__init__() call"
                 raise AttributeError(msg)