diff --git a/backend/src/InterpolateTorch.py b/backend/src/InterpolateTorch.py index 9f30f573..681d3528 100644 --- a/backend/src/InterpolateTorch.py +++ b/backend/src/InterpolateTorch.py @@ -13,9 +13,9 @@ log ) from .version import __version__ -torch.set_float32_matmul_precision("high") +torch.set_float32_matmul_precision("medium") torch.set_grad_enabled(False) - +logging.basicConfig(level=logging.INFO) class InterpolateRifeTorch: """InterpolateRifeTorch class for video interpolation using RIFE model in PyTorch. @@ -289,7 +289,7 @@ def __init__( if self.backend == "tensorrt": import tensorrt import torch_tensorrt - logging.basicConfig(level=logging.ERROR) + logging.basicConfig(level=logging.INFO) trt_engine_path = os.path.join( os.path.realpath(trt_cache_dir), ( diff --git a/src/Util.py b/src/Util.py index 49704a38..1c52fe06 100644 --- a/src/Util.py +++ b/src/Util.py @@ -267,7 +267,7 @@ def downloadFile(link, downloadLocation): ) with open(downloadLocation, "wb") as f: - for chunk in response.iter_content(chunk_size=128): + for chunk in response.iter_content(chunk_size=1024): f.write(chunk)