Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove unused helper (since LoRA refactoring)

Browse source
This commit is contained in:
Pierre Chapuis 2024-01-29 16:31:58 +01:00 committed by Cédric Deltheil
parent ae19892d1d
commit c57f2228f8
1 changed files with 0 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -198,11 +198,6 @@ def load_from_safetensors(path: Path | str, device: Device | str = "cpu") -> dic
return {key: tensors.get_tensor(key) for key in tensors.keys()} # type: ignore return {key: tensors.get_tensor(key) for key in tensors.keys()} # type: ignore
def load_metadata_from_safetensors(path: Path | str) -> dict[str, str] | None:
with safe_open(path=path, framework="pytorch") as tensors: # type: ignore
return tensors.metadata() # type: ignore
def save_to_safetensors(path: Path | str, tensors: dict[str, Tensor], metadata: dict[str, str] | None = None) -> None: def save_to_safetensors(path: Path | str, tensors: dict[str, Tensor], metadata: dict[str, str] | None = None) -> None:
_save_file(tensors, path, metadata) # type: ignore _save_file(tensors, path, metadata) # type: ignore