diff --git a/MANIFEST.in b/MANIFEST.in index fd5bfe2..90d8702 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -1 +1,2 @@ recursive-include scepter *.yaml +recursive-include scepter *.md diff --git a/readme.md b/readme.md index 5720dc0..a401d85 100644 --- a/readme.md +++ b/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) | diff --git a/scepter/modules/inference/diffusion_inference.py b/scepter/modules/inference/diffusion_inference.py index 0528ec9..2b7314b 100644 --- a/scepter/modules/inference/diffusion_inference.py +++ b/scepter/modules/inference/diffusion_inference.py @@ -255,8 +255,11 @@ class DiffusionInference(): 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 diff --git a/scepter/studio/tuner_manager/manager_ui/browser_ui.py b/scepter/studio/tuner_manager/manager_ui/browser_ui.py index e930b19..b0f21bc 100644 --- a/scepter/studio/tuner_manager/manager_ui/browser_ui.py +++ b/scepter/studio/tuner_manager/manager_ui/browser_ui.py @@ -6,6 +6,7 @@ from collections import OrderedDict 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 @@ class BrowserUI(UIBase): 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() diff --git a/scepter/version.py b/scepter/version.py index e588539..230866d 100644 --- a/scepter/version.py +++ b/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])