🔨 chore(generate_image): update cfg_rescale parameter type to float

LlmKira · Feb 14, 2024 · 0b22334 · 0b22334
1 parent e24a7f1
commit 0b22334
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/novelai_python/sdk/ai/generate_image/__init__.py b/src/novelai_python/sdk/ai/generate_image/__init__.py
@@ -236,7 +236,7 @@ def build(cls,
               seed: int = None,
               steps: int = 28,
               scale: float = 6.0,
-              cfg_rescale: int = 0,
+              cfg_rescale: float = 0,
               sampler: Union[Sampler, str] = Sampler.K_EULER,
               width: int = 832,
               height: int = 1216,