update v0.0.5.post1

This commit is contained in:
靖渊
2024-04-29 16:02:23 +08:00
parent f502f8063d
commit 6be496d5ac
5 changed files with 11 additions and 3 deletions
@@ -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
@@ -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()
+1 -1
View File
@@ -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])