17 lines
476 B
Python
17 lines
476 B
Python
def test_load_model():
|
|
from inpaint.plugins import InteractiveSeg
|
|
from inpaint.model_manager import ModelManager
|
|
|
|
interactive_seg_model = InteractiveSeg("vit_l", "cpu")
|
|
|
|
models = ["lama", "ldm", "zits", "mat", "fcf", "manga", "migan"]
|
|
for m in models:
|
|
ModelManager(
|
|
name=m,
|
|
device="cpu",
|
|
no_half=False,
|
|
disable_nsfw=False,
|
|
sd_cpu_textencoder=True,
|
|
cpu_offload=True,
|
|
)
|