mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
21 lines
575 B
Python
21 lines
575 B
Python
import argparse
|
|
import torch
|
|
import os
|
|
from diffusers import StableDiffusionPipeline
|
|
import sys
|
|
|
|
PROJECT_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
# add project root to path
|
|
sys.path.append(PROJECT_ROOT)
|
|
|
|
SAMPLER_SCALES_ROOT = os.path.join(PROJECT_ROOT, 'toolkit', 'samplers_scales')
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Process some images.')
|
|
add_arg = parser.add_argument
|
|
add_arg('--model', type=str, required=True, help='Path to model')
|
|
add_arg('--sampler', type=str, required=True, help='Name of sampler')
|
|
|
|
args = parser.parse_args()
|
|
|