fix noise map

server/pipelines/pix2pixTurbo.py

@@ -49,16 +49,20 @@ class Pipeline:
         height: int = Field(
             512, min=2, max=15, title="Height", disabled=True, hide=True, id="height"
         )
-        strength: float = Field(
+        seed: int = Field(
+            2159232, min=0, title="Seed", field="seed", hide=True, id="seed"
+        )
+        noise_r: float = Field(
             1.0,
             min=0.01,
-            max=10.0,
+            max=3.0,
             step=0.001,
-            title="Strength",
+            title="Noise R",
             field="range",
             hide=True,
-            id="strength",
+            id="noise_r",
         )
+
         deterministic: bool = Field(
             True,
             hide=True,
@@ -107,12 +111,17 @@ class Pipeline:
             params.canny_high_threshold,
             output_type="pil,tensor",
         )
+        torch.manual_seed(params.seed)
+        noise = torch.randn(
+            (1, 4, params.width // 8, params.height // 8), device=self.device
+        )
         canny_tensor = torch.cat((canny_tensor, canny_tensor, canny_tensor), dim=1)
         output_image = self.model(
             canny_tensor,
             params.prompt,
             params.deterministic,
-            params.strength,
+            params.noise_r,
+            noise,
         )
         output_pil = transforms.ToPILImage()(output_image[0].cpu() * 0.5 + 0.5)
 

server/requirements.txt

@@ -2,7 +2,7 @@ diffusers==0.25.1
 transformers==4.36.2
 --extra-index-url https://download.pytorch.org/whl/cu121;
 torch==2.2.0
-fastapi==0.108.0
+fastapi==0.110.0
 uvicorn[standard]==0.25.0
 Pillow==10.2.0
 accelerate==0.25.0
@@ -15,4 +15,6 @@ safetensors
 stable_fast @ https://github.com/chengzeyi/stable-fast/releases/download/v1.0.4/stable_fast-1.0.4+torch220cu121-cp310-cp310-manylinux2014_x86_64.whl ; sys_platform != 'darwin' or platform_machine != 'arm64'
 oneflow @ https://github.com/siliconflow/oneflow_releases/releases/download/community_cu121/oneflow-0.9.1.dev20240316+cu121-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl ; sys_platform != 'darwin' or platform_machine != 'arm64'
 onediff @ git+https://github.com/siliconflow/onediff.git@main#egg=onediff ; sys_platform != 'darwin' or platform_machine != 'arm64'
-setuptools
+setuptools
+mpmath==1.3.0
+numpy==1.*