Function load_pt

src/diffusers/utils/testing_utils.py:708–712

(url: str, map_location: str | None = None, weights_only: bool | None = True)

def load_pt(url: str, map_location: str | None = None, weights_only: bool | None = True):
    response = requests.get(url, timeout=DIFFUSERS_REQUEST_TIMEOUT)
    response.raise_for_status()
    arry = torch.load(BytesIO(response.content), map_location=map_location, weights_only=weights_only)
    return arry


def load_image(image: str | PIL.Image.Image) -> PIL.Image.Image: