Spaces:
Running
on
Zero
Running
on
Zero
chaojiemao
commited on
Commit
•
161b0b1
1
Parent(s):
0f38f9c
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ import transformers
|
|
24 |
from PIL import Image
|
25 |
from transformers import AutoModel, AutoTokenizer
|
26 |
|
27 |
-
from
|
28 |
from scepter.modules.utils.config import Config
|
29 |
from scepter.modules.utils.directory import get_md5
|
30 |
from scepter.modules.utils.file_system import FS
|
@@ -90,7 +90,7 @@ class ChatBotUI(object):
|
|
90 |
#FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer/")
|
91 |
|
92 |
assert len(self.model_choices) > 0
|
93 |
-
if self.default_model_name == "": self.default_model_name = self.model_choices.keys()[0]
|
94 |
self.model_name = self.default_model_name
|
95 |
self.pipe = ACEInference()
|
96 |
self.pipe.init_from_cfg(self.model_choices[self.default_model_name])
|
|
|
24 |
from PIL import Image
|
25 |
from transformers import AutoModel, AutoTokenizer
|
26 |
|
27 |
+
from ace_inference import ACEInference
|
28 |
from scepter.modules.utils.config import Config
|
29 |
from scepter.modules.utils.directory import get_md5
|
30 |
from scepter.modules.utils.file_system import FS
|
|
|
90 |
#FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer/")
|
91 |
|
92 |
assert len(self.model_choices) > 0
|
93 |
+
if self.default_model_name == "": self.default_model_name = list(self.model_choices.keys())[0]
|
94 |
self.model_name = self.default_model_name
|
95 |
self.pipe = ACEInference()
|
96 |
self.pipe.init_from_cfg(self.model_choices[self.default_model_name])
|