File size: 476 Bytes
cbbdd92 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
def test_load_model():
from iopaint.plugins import InteractiveSeg
from iopaint.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,
)
|