s

TNTwise · Sep 9, 2024 · 97cc1fa · 97cc1fa
1 parent 2039614
commit 97cc1fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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)