1fee098f10
Signed-off-by: <> Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com> Co-authored-by: Will Lin <wlsaidhi@gmail.com> Co-authored-by: Zhou, Wei <wzhou322@gatech.edu> Co-authored-by: Kevin Lin <42618777+kevin314@users.noreply.github.com> Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com> Co-authored-by: JerryZhou54 <69577934+JerryZhou54@users.noreply.github.com> Co-authored-by: Yongqi Chen <144848849+BrianChen1129@users.noreply.github.com> Co-authored-by: Peiyuan Zhang <m2deng@ucsd.edu>
19 lines
460 B
Python
19 lines
460 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""Utilities for selecting and loading models."""
|
|
import contextlib
|
|
|
|
import torch
|
|
|
|
from fastvideo.v1.logger import init_logger
|
|
|
|
logger = init_logger(__name__)
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def set_default_torch_dtype(dtype: torch.dtype):
|
|
"""Sets the default torch dtype to the given dtype."""
|
|
old_dtype = torch.get_default_dtype()
|
|
torch.set_default_dtype(dtype)
|
|
yield
|
|
torch.set_default_dtype(old_dtype)
|