update v0.0.5.post1
This commit is contained in:
@@ -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
@@ -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])
|
||||
|
||||
|
||||
Reference in New Issue
Block a user