update

ootd/inference_ootd_dc.py

@@ -56,12 +56,13 @@ class OOTDiffusionDC:
 
         #判断文件是否存在
         filePath = "/home/user/app/checkpoints/ootd/text_encoder/pytorch_model.bin"
-        if os.path.exists(filePath) == False:
+        if not os.path.exists(filePath) :
             url = "https://huggingface.co/yangjoe/pytorch_model/resolve/main/pytorch_model.bin"
             response = requests.get(url)
             #下载该文件
             with open("/home/user/app/checkpoints/ootd/text_encoder/pytorch_model.bin", "wb") as f:
                 f.write(response.content)
+            print("pytorch_model.bin下载完成")
 
         self.pipe = OotdPipeline.from_pretrained(
             MODEL_PATH,

ootd/inference_ootd_hd.py

@@ -56,12 +56,13 @@ class OOTDiffusionHD:
         )
         #判断文件是否存在
         filePath = "/home/user/app/checkpoints/ootd/text_encoder/pytorch_model.bin"
-        if os.path.exists(filePath) == False:
+        if not os.path.exists(filePath) :
             url = "https://huggingface.co/yangjoe/pytorch_model/resolve/main/pytorch_model.bin"
             response = requests.get(url)
             #下载该文件
             with open("/home/user/app/checkpoints/ootd/text_encoder/pytorch_model.bin", "wb") as f:
                 f.write(response.content)
+            print("pytorch_model.bin下载完成")
 
         self.pipe = OotdPipeline.from_pretrained(
             MODEL_PATH,

preprocess/openpose/annotator/openpose/__init__.py

@@ -16,6 +16,7 @@ from .body import Body
 from .hand import Hand
 from .face import Face
 from annotator.util import annotator_ckpts_path
+import requests
 
 body_model_path = "https://huggingface.co/lllyasviel/Annotators/resolve/main/body_pose_model.pth"
 hand_model_path = "https://huggingface.co/lllyasviel/Annotators/resolve/main/hand_pose_model.pth"
@@ -49,8 +50,13 @@ class OpenposeDetector:
         # face_modelpath = os.path.join(annotator_ckpts_path, "facenet.pth")
 
         if not os.path.exists(body_modelpath):
-            from basicsr.utils.download_util import load_file_from_url
-            load_file_from_url(body_model_path, model_dir=annotator_ckpts_path)
+            # from basicsr.utils.download_util import load_file_from_url
+            # load_file_from_url(body_model_path, model_dir=annotator_ckpts_path)
+            response = requests.get(body_model_path)
+            #下载该文件
+            with open("/home/user/app/checkpoints/openpose/ckpts/body_pose_model.pth", "wb") as f:
+                f.write(response.content)
+            print("body_pose_model.pth下载完成")
 
         # if not os.path.exists(hand_modelpath):
         #     from basicsr.utils.download_util import load_file_from_url