stable-diffusion-webui

1
from __future__ import annotations
2

3
import torch.nn
4

5

6
def get_param(model) -> torch.nn.Parameter:
7
    """
8
    Find the first parameter in a model or module.
9
    """
10
    if hasattr(model, "model") and hasattr(model.model, "parameters"):
11
        # Unpeel a model descriptor to get at the actual Torch module.
12
        model = model.model
13

14
    for param in model.parameters():
15
        return param
16

17
    raise ValueError(f"No parameters found in model {model!r}")
18