mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 23:00:51 +08:00
38 lines
2.2 KiB
Python
38 lines
2.2 KiB
Python
from singleton_decorator import singleton
|
|
import os
|
|
|
|
@singleton
|
|
class PathServer():
|
|
|
|
def __init__(self):
|
|
self.paths = {
|
|
'checkpoints': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'checkpoints'),
|
|
'clip': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'clip'),
|
|
'clip_vision': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'clip_vision'),
|
|
'configs': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'configs'),
|
|
'controlnet': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'controlnet'),
|
|
'embeddings': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'embeddings'),
|
|
'loras': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'loras'),
|
|
'style_models': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'style_models'),
|
|
't2i_adapter': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 't2i_adapter'),
|
|
'upscale_models': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'upscale_models'),
|
|
'vae': os.path.join(os.path.dirname(os.path.realpath(__file__)), 'models', 'vae'),
|
|
}
|
|
|
|
def set_a1111_path(self, a1111_path):
|
|
self.paths['checkpoints'] = os.path.join(a1111_path, 'models', 'Stable-diffusion')
|
|
self.paths['clip'] = os.path.join(a1111_path, 'models', 'clip-interrogator')
|
|
# self.paths['clip_vision'] = os.path.join(a1111_path, 'models', '')
|
|
self.paths['configs'] = os.path.join(a1111_path, 'models', 'Stable-diffusion')
|
|
self.paths['controlnet'] = os.path.join(a1111_path, 'models', 'ControlNet')
|
|
self.paths['embeddings'] = os.path.join(a1111_path, 'embeddings')
|
|
self.paths['loras'] = os.path.join(a1111_path, 'models', 'Lora')
|
|
#self.paths['style_models'] = os.path.join(a1111_path, 'models', '')
|
|
#self.paths['t2i_adapter'] = os.path.join(a1111_path, 'models', '')
|
|
self.paths['upscale_models'] = os.path.join(a1111_path, 'models', 'ESRGAN')
|
|
self.paths['vae'] = os.path.join(a1111_path, 'models', 'VAE')
|
|
|
|
def get(self, key):
|
|
return self.paths[key]
|
|
|