Add ruff workflow and pyproject

MisterChief95 · Nov 21, 2024 · d547bd0 · d547bd0
1 parent a939f62
commit d547bd0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/hidiffusion/raunet.py b/hidiffusion/raunet.py
@@ -1,11 +1,19 @@
 import os
 from typing import Any
 
+import torch
 import torch.nn.functional as F
 
 import backend.nn.unet as unet
 
-from .utils import *
+from .utils import (
+    check_time,
+    convert_time,
+    get_sigma,
+    parse_blocks,
+    scale_samples,
+    UPSCALE_METHODS,
+)
 from .logger import logger