update

Author: hlky

src/diffusers/utils/constants.py

@@ -56,3 +56,14 @@
 
 if USE_PEFT_BACKEND and _CHECK_PEFT:
     dep_version_check("peft")
+
+
+DECODE_ENDPOINT_SD_V1 = "https://q1bj3bpq6kzilnsu.us-east-1.aws.endpoints.huggingface.cloud/"
+DECODE_ENDPOINT_SD_XL = "https://x2dmsqunjd6k9prw.us-east-1.aws.endpoints.huggingface.cloud/"
+DECODE_ENDPOINT_FLUX = "https://whhx50ex1aryqvw6.us-east-1.aws.endpoints.huggingface.cloud/"
+DECODE_ENDPOINT_HUNYUAN_VIDEO = "https://o7ywnmrahorts457.us-east-1.aws.endpoints.huggingface.cloud/"
+
+
+ENCODE_ENDPOINT_SD_V1 = "https://qc6479g0aac6qwy9.us-east-1.aws.endpoints.huggingface.cloud/"
+ENCODE_ENDPOINT_SD_XL = "https://xjqqhmyn62rog84g.us-east-1.aws.endpoints.huggingface.cloud/"
+ENCODE_ENDPOINT_FLUX = "https://ptccx55jz97f9zgo.us-east-1.aws.endpoints.huggingface.cloud/"

src/diffusers/utils/remote_utils.py

@@ -43,17 +43,6 @@
 from PIL import Image
 
 
-DECODE_ENDPOINT_SD_V1 = "https://q1bj3bpq6kzilnsu.us-east-1.aws.endpoints.huggingface.cloud/"
-DECODE_ENDPOINT_SD_XL = "https://x2dmsqunjd6k9prw.us-east-1.aws.endpoints.huggingface.cloud/"
-DECODE_ENDPOINT_FLUX = "https://whhx50ex1aryqvw6.us-east-1.aws.endpoints.huggingface.cloud/"
-DECODE_ENDPOINT_HUNYUAN_VIDEO = "https://o7ywnmrahorts457.us-east-1.aws.endpoints.huggingface.cloud/"
-
-
-ENCODE_ENDPOINT_SD_V1 = "https://qc6479g0aac6qwy9.us-east-1.aws.endpoints.huggingface.cloud/"
-ENCODE_ENDPOINT_SD_XL = "https://xjqqhmyn62rog84g.us-east-1.aws.endpoints.huggingface.cloud/"
-ENCODE_ENDPOINT_FLUX = "https://ptccx55jz97f9zgo.us-east-1.aws.endpoints.huggingface.cloud/"
-
-
 def detect_image_type(data: bytes) -> str:
     if data.startswith(b"\xff\xd8"):
         return "jpeg"

tests/remote/test_remote_decode.py

@@ -21,11 +21,13 @@
 import torch
 
 from diffusers.image_processor import VaeImageProcessor
-from diffusers.utils.remote_utils import (
+from diffusers.utils.constants import (
     DECODE_ENDPOINT_FLUX,
     DECODE_ENDPOINT_HUNYUAN_VIDEO,
     DECODE_ENDPOINT_SD_V1,
     DECODE_ENDPOINT_SD_XL,
+)
+from diffusers.utils.remote_utils import (
     remote_decode,
 )
 from diffusers.utils.testing_utils import (

tests/remote/test_remote_encode.py

@@ -19,13 +19,15 @@
 import torch
 
 from diffusers.utils import load_image
-from diffusers.utils.remote_utils import (
+from diffusers.utils.constants import (
     DECODE_ENDPOINT_FLUX,
     DECODE_ENDPOINT_SD_V1,
     DECODE_ENDPOINT_SD_XL,
     ENCODE_ENDPOINT_FLUX,
     ENCODE_ENDPOINT_SD_V1,
     ENCODE_ENDPOINT_SD_XL,
+)
+from diffusers.utils.remote_utils import (
     remote_decode,
     remote_encode,
 )