Spaces:
Running
Running
| 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_disable_nsfw: bool = False | |
| sd_cpu_textencoder: bool = True | |
| sd_run_local: bool = False | |
| sd_enable_xformers: bool = False | |
| local_files_only: bool = False | |
| cpu_offload: bool = False | |
| device: str = "cuda" | |
| 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 | |
| model_dir: str = None | |
| output_dir: str = None | |
| if __name__ == "__main__": | |
| main(FakeArgs()) | |