Be more clear about Spandrel model nomenclature

This commit is contained in:
Aarni Koskela
2023-12-31 00:09:51 +02:00
parent c0ca6348e8
commit 777af661a2
5 changed files with 13 additions and 11 deletions

View File

@@ -143,7 +143,7 @@ def load_spandrel_model(
*,
device: str | torch.device | None,
half: bool = False,
dtype: str | None = None,
dtype: str | torch.dtype | None = None,
expected_architecture: str | None = None,
) -> spandrel.ModelDescriptor:
import spandrel