strips/stable-diffusion-rpc
2
0
Fork 0
mirror of https://github.com/GayPizzaSpecifications/stable-diffusion-rpc.git synced 2025-08-04 14:01:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add command-line options for bind host and port.

Browse Source
This commit is contained in:
Alex Zenla
2023-04-23 03:56:55 -07:00
parent d31e80bf4c
commit 1bb629c18f
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Sources/StableDiffusionServer/main.swift
View File

@ -9,6 +9,12 @@ struct ServerCommand: ParsableCommand {
@Option(name: .shortAndLong, help: "Path to models directory") @Option(name: .shortAndLong, help: "Path to models directory")
var modelsDirectoryPath: String = "models" var modelsDirectoryPath: String = "models"
@Option(name: .long, help: "Bind host")
var bindHost: String = "0.0.0.0"
@Option(name: .long, help: "Bind port")
var bindPort: Int = 4546
mutating func run() throws { mutating func run() throws {
let modelsDirectoryURL = URL(filePath: modelsDirectoryPath) let modelsDirectoryURL = URL(filePath: modelsDirectoryPath)
let modelManager = ModelManager(modelBaseURL: modelsDirectoryURL) let modelManager = ModelManager(modelBaseURL: modelsDirectoryURL)
@ -30,7 +36,7 @@ struct ServerCommand: ParsableCommand {
ModelServiceProvider(modelManager: modelManager), ModelServiceProvider(modelManager: modelManager),
ImageGenerationServiceProvider(modelManager: modelManager) ImageGenerationServiceProvider(modelManager: modelManager)
]) ])
.bind(host: "0.0.0.0", port: 4546) .bind(host: bindHost, port: bindPort)
dispatchMain() dispatchMain()
} }