| from typing import List |
| from pydantic import BaseModel |
| from lama_cleaner.server import main |
|
|
| class FakeArgs(BaseModel): |
| host: str = "0.0.0.0" |
| port: int = 7860 |
| model: str = 'lama' |
| hf_access_token: str = "" |
| sd_enable_xformers: bool = False |
| sd_disable_nsfw: bool = False |
| sd_cpu_textencoder: bool = True |
| sd_controlnet: bool = False |
| sd_controlnet_method: str = "control_v11p_sd15_canny" |
| sd_local_model_path: str = "" |
| sd_run_local: bool = False |
| local_files_only: bool = False |
| cpu_offload: bool = False |
| device: str = "cpu" |
| gui: bool = False |
| gui_size: List[int] = [1000, 1000] |
| input: str = '' |
| disable_model_switch: bool = True |
| debug: bool = False |
| no_half: bool = False |
| disable_nsfw: bool = False |
| enable_xformers: bool = False |
| enable_interactive_seg: bool = True |
| interactive_seg_model: str = "vit_b" |
| interactive_seg_device: str = "cpu" |
| enable_remove_bg: bool = False |
| enable_anime_seg: bool = False |
| enable_realesrgan: bool = False |
| enable_gfpgan: bool = False |
| gfpgan_device: str = "cpu" |
| enable_restoreformer: bool = False |
| enable_gif: bool = False |
| quality: int = 95 |
| model_dir: str = None |
| output_dir: str = None |
|
|
| if __name__ == "__main__": |
| main(FakeArgs()) |
|
|
