Merge pull request #28 from modelscope/v0.0.5_dev

V0.0.5 dev

MANIFEST.in

@@ -1 +1,2 @@
 recursive-include scepter *.yaml
+recursive-include scepter *.md

readme.md

@@ -132,6 +132,8 @@ The startup of **SCEPTER Studio** eliminates the need for manual downloading and
 Depending on the network and hardware situation, the initial startup usually requires 15-60 minutes, primarily involving the download and processing of SDv1.5, SDv2.1, and SDXL models.
 Therefore, subsequent startups will become much faster (about one minute) as downloading is no longer required.
 
+To support the sharing and downloading of models,
+please make sure that you have installed **zip** and Git Large File Storage (**git lfs**).
 ### Usage Demo
 
 |              [Image Editing](https://www.modelscope.cn/api/v1/models/iic/scepter/repo?Revision=master&FilePath=assets%2Fscepter_studio%2Fimage_editing_20240419.webm)              |                [Training](https://www.modelscope.cn/api/v1/models/iic/scepter/repo?Revision=master&FilePath=assets%2Fscepter_studio%2Ftraining_20240419.webm)                 |              [Model Sharing](https://www.modelscope.cn/api/v1/models/iic/scepter/repo?Revision=master&FilePath=assets%2Fscepter_studio%2Fmodel_sharing_20240419.webm)               |             [Model Inference](https://www.modelscope.cn/api/v1/models/iic/scepter/repo?Revision=master&FilePath=assets%2Fscepter_studio%2Fmodel_inference_20240419.webm)              |             [Data Management](https://www.modelscope.cn/api/v1/models/iic/scepter/repo?Revision=master&FilePath=assets%2Fscepter_studio%2Fdata_management_20240419.webm)              |

scepter/modules/inference/diffusion_inference.py

@@ -255,8 +255,11 @@ def unload(self, module):
             module['device'] = 'offline'
             print('delete module')
         else:
-            module['model'] = module['model'].to('cpu')
-            module['device'] = 'cpu'
+            if module['model'] is not None:
+                module['model'] = module['model'].to('cpu')
+                module['device'] = 'cpu'
+            else:
+                module['device'] = 'offline'
         torch.cuda.empty_cache()
         torch.cuda.ipc_collect()
         return module

scepter/studio/tuner_manager/manager_ui/browser_ui.py

@@ -6,6 +6,7 @@
 import gradio as gr
 from swift import push_to_hub
 
+import scepter
 from scepter.modules.utils.config import Config
 from scepter.modules.utils.file_system import FS
 from scepter.studio.tuner_manager.manager_ui.component_names import \
@@ -36,6 +37,7 @@ def __init__(self, cfg, language='en'):
         self.language = language
         self.export_folder = os.path.join(self.work_dir, cfg.EXPORT_DIR)
         self.readme_file = cfg.README_EN if self.language == 'en' else cfg.README_ZH
+        self.readme_file = os.path.join(os.path.dirname(scepter.dirname), self.readme_file)
 
     def saved_tuners_to_category(self):
         self.saved_tuners_category = OrderedDict()

scepter/version.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Copyright (c) Alibaba, Inc. and its affiliates.
 
-__version__ = '0.0.5'
+__version__ = '0.0.5.post1'
 
 version_info = tuple(int(x) for x in __version__.split('.')[0:3])