fix error

.gitignore

@@ -0,0 +1,20 @@
+*.pyc
+*.pth
+*.pt
+*.pkl
+*.ckpt
+*.DS_Store
+*__pycache__*
+*.cache*
+*.bin
+*.idea
+*.csv
+cache
+build
+dist
+dev
+scepter.egg-info
+.readthedocs.yml
+*resources
+*.ipynb_checkpoints*
+*.vscode

app.py

@@ -27,9 +27,9 @@ from scepter.modules.utils.directory import get_md5
 from scepter.modules.utils.file_system import FS
 from scepter.studio.utils.env import init_env
 
-from .infer import ACEInference
-from .example import get_examples
-from .utils import load_image
+from infer import ACEInference
+from example import get_examples
+from utils import load_image
 
 
 refresh_sty = '\U0001f504'  # 🔄
@@ -44,10 +44,9 @@ lock = threading.Lock()
 
 class ChatBotUI(object):
     def __init__(self,
-                 cfg_general_file,
+                 cfg,
                  root_work_dir='./'):
 
-        cfg = Config(cfg_file=cfg_general_file)
         cfg.WORK_DIR = os.path.join(root_work_dir, cfg.WORK_DIR)
         if not FS.exists(cfg.WORK_DIR):
             FS.make_dir(cfg.WORK_DIR)
@@ -1189,11 +1188,10 @@ class ChatBotUI(object):
 
 
 if __name__ == '__main__':
-    cfg = Config(cfg_file="config/chatbot_ui.yaml")
 
+    cfg = Config(cfg_file="config/chatbot_ui.yaml")
     with gr.Blocks() as demo:
         chatbot = ChatBotUI(cfg)
-        chatbot.create_bot_ui()
+        chatbot.create_ui()
         chatbot.set_callbacks()
-    
     demo.launch()

config/chatbot_ui.yaml

@@ -2,10 +2,12 @@ WORK_DIR: ./cache/chatbot
 FILE_SYSTEM:
   - NAME: LocalFs
     TEMP_DIR: ./cache
-  - NAME: ModelscopeFs
+  - NAME: HttpFs
     TEMP_DIR: ./cache
   - NAME: HuggingfaceFs
     TEMP_DIR: ./cache
+  - NAME: ModelscopeFs
+    TEMP_DIR: ./cache
 #
 ENABLE_I2V: False
 #

infer.py

@@ -12,17 +12,48 @@ import torch.nn.functional as F
 import torchvision.transforms.functional as TF
 
 from scepter.modules.model.registry import DIFFUSIONS
-from scepter.modules.model.utils.basic_utils import (
-    check_list_of_list,
-    pack_imagelist_into_tensor_v2 as pack_imagelist_into_tensor,
-    to_device,
-    unpack_tensor_into_imagelist
-)
 from scepter.modules.utils.distribute import we
 from scepter.modules.utils.logger import get_logger
-
 from scepter.modules.inference.diffusion_inference import DiffusionInference, get_model
 
+def check_list_of_list(ll):
+    return isinstance(ll, list) and all(isinstance(i, list) for i in ll)
+
+def pack_imagelist_into_tensor(image_list):
+    # allow None
+    example = None
+    image_tensor, shapes = [], []
+    for img in image_list:
+        if img is None:
+            example = find_example(image_tensor,
+                                   image_list) if example is None else example
+            image_tensor.append(example)
+            shapes.append(None)
+            continue
+        _, c, h, w = img.size()
+        image_tensor.append(img.view(c, h * w).transpose(1, 0))  # h*w, c
+        shapes.append((h, w))
+
+    image_tensor = pad_sequence(image_tensor,
+                                batch_first=True).permute(0, 2, 1)  # b, c, l
+    return image_tensor, shapes
+
+def to_device(inputs, strict=True):
+    if inputs is None:
+        return None
+    if strict:
+        assert all(isinstance(i, torch.Tensor) for i in inputs)
+    return [i.to(we.device_id) if i is not None else None for i in inputs]
+
+
+def unpack_tensor_into_imagelist(image_tensor, shapes):
+    image_list = []
+    for img, shape in zip(image_tensor, shapes):
+        h, w = shape[0], shape[1]
+        image_list.append(img[:, :h * w].view(1, -1, h, w))
+
+    return image_list
+
 
 def process_edit_image(images,
                        masks,

requirements.txt

@@ -1,5 +1,7 @@
-huggingface_hub==0.25.2
+huggingface_hub
 scepter>=1.1.0
 diffusers
 transformers
-gradio_imageslider
+gradio_imageslider
+torch>=2.4.0
+torchvision