wolverinn commited on
Commit
99ab15d
·
1 Parent(s): e516e30

fix import

Browse files
handler.py CHANGED
@@ -13,8 +13,7 @@ from packaging import version
13
  import logging
14
  logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
15
 
16
- from modules import import_hook, errors, extra_networks, ui_extra_networks_checkpoints
17
- from modules import extra_networks_hypernet, ui_extra_networks_hypernets, ui_extra_networks_textual_inversion
18
  from modules.call_queue import wrap_queued_call, queue_lock, wrap_gradio_gpu_call
19
 
20
  import torch
@@ -24,7 +23,7 @@ if ".dev" in torch.__version__ or "+git" in torch.__version__:
24
  torch.__long_version__ = torch.__version__
25
  torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
26
 
27
- from modules import shared, devices, sd_samplers, upscaler, extensions, localization, ui_tempdir, ui_extra_networks
28
  import modules.codeformer_model as codeformer
29
  import modules.face_restoration
30
  import modules.gfpgan_model as gfpgan
 
13
  import logging
14
  logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
15
 
16
+ from modules import errors
 
17
  from modules.call_queue import wrap_queued_call, queue_lock, wrap_gradio_gpu_call
18
 
19
  import torch
 
23
  torch.__long_version__ = torch.__version__
24
  torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)
25
 
26
+ from modules import shared, devices, ui_tempdir
27
  import modules.codeformer_model as codeformer
28
  import modules.face_restoration
29
  import modules.gfpgan_model as gfpgan
modules/ui_extra_networks_checkpoints.py CHANGED
@@ -1,39 +1,39 @@
1
- import html
2
- import json
3
- import os
4
- import urllib.parse
5
-
6
- from modules import shared, ui_extra_networks, sd_models
7
-
8
-
9
- class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
10
- def __init__(self):
11
- super().__init__('Checkpoints')
12
-
13
- def refresh(self):
14
- shared.refresh_checkpoints()
15
-
16
- def list_items(self):
17
- checkpoint: sd_models.CheckpointInfo
18
- for name, checkpoint in sd_models.checkpoints_list.items():
19
- path, ext = os.path.splitext(checkpoint.filename)
20
- previews = [path + ".png", path + ".preview.png"]
21
-
22
- preview = None
23
- for file in previews:
24
- if os.path.isfile(file):
25
- preview = self.link_preview(file)
26
- break
27
-
28
- yield {
29
- "name": checkpoint.name_for_extra,
30
- "filename": path,
31
- "preview": preview,
32
- "search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),
33
- "onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"',
34
- "local_preview": path + ".png",
35
- }
36
-
37
- def allowed_directories_for_previews(self):
38
- return [v for v in [shared.cmd_opts.ckpt_dir, sd_models.model_path] if v is not None]
39
 
 
1
+ # import html
2
+ # import json
3
+ # import os
4
+ # import urllib.parse
5
+
6
+ # from modules import shared, ui_extra_networks, sd_models
7
+
8
+
9
+ # class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
10
+ # def __init__(self):
11
+ # super().__init__('Checkpoints')
12
+
13
+ # def refresh(self):
14
+ # shared.refresh_checkpoints()
15
+
16
+ # def list_items(self):
17
+ # checkpoint: sd_models.CheckpointInfo
18
+ # for name, checkpoint in sd_models.checkpoints_list.items():
19
+ # path, ext = os.path.splitext(checkpoint.filename)
20
+ # previews = [path + ".png", path + ".preview.png"]
21
+
22
+ # preview = None
23
+ # for file in previews:
24
+ # if os.path.isfile(file):
25
+ # preview = self.link_preview(file)
26
+ # break
27
+
28
+ # yield {
29
+ # "name": checkpoint.name_for_extra,
30
+ # "filename": path,
31
+ # "preview": preview,
32
+ # "search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),
33
+ # "onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"',
34
+ # "local_preview": path + ".png",
35
+ # }
36
+
37
+ # def allowed_directories_for_previews(self):
38
+ # return [v for v in [shared.cmd_opts.ckpt_dir, sd_models.model_path] if v is not None]
39