File size: 273 Bytes
3dabe4a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from dataclasses import dataclass
from torch import Tensor


@dataclass
class ReSharpenParams:

    enable: bool = False
    scaling: str = None
    strength: float = 0.0
    total_step: int = -1
    cache: Tensor = None

    def __bool__(self):
        return self.enable