Files
ZHO-ZHO-ZHO-ComfyUI-SegMoE/segmoe/cli.py
T
2024-02-05 03:28:55 +08:00

8 lines
142 B
Python

from .main import SegMoEPipeline
import sys
def create(args=sys.argv):
pipe = SegMoEPipeline(args[1])
pipe.save_pretrained(args[2])