Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

deprecate bidirectional_mapping util

Browse source
This commit is contained in:
Cédric Deltheil 2023-09-07 18:18:19 +02:00 committed by Cédric Deltheil
parent 2786117469
commit c6fadd1c81
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/refiners/fluxion/utils.py
View file

@ -1,4 +1,4 @@
from typing import Dict, Iterable, Literal, TypeVar
from typing import Iterable, Literal, TypeVar
from PIL import Image
from numpy import array, float32
from pathlib import Path
@ -34,10 +34,6 @@ def interpolate(x: Tensor, factor: float | torch.Size, mode: str = "nearest") ->
) # type: ignore
def bidirectional_mapping(mapping: Dict[str, str]) -> Dict[str, str]:
return {**mapping, **{value: key for key, value in mapping.items()}}
def image_to_tensor(image: Image.Image, device: Device | str | None = None, dtype: DType | None = None) -> Tensor:
return torch.tensor(array(image).astype(float32).transpose(2, 0, 1) / 255.0, device=device, dtype=dtype).unsqueeze(
0