mirror of
https://github.com/karl0ss/ai_image_frame_server.git
synced 2025-06-08 03:25:06 +01:00
Compare commits
2 Commits
c4b9dd2921
...
7cf3fdb432
Author | SHA1 | Date | |
---|---|---|---|
![]() |
7cf3fdb432 | ||
![]() |
32afc70d03 |
15
lib.py
15
lib.py
@ -199,15 +199,22 @@ def generate_image(
|
||||
|
||||
# Conditionally set model if node and param are provided
|
||||
if model_node and model_param:
|
||||
if user_config["comfyui"].get("FLUX"):
|
||||
available_model_list = user_config["comfyui"]["confyui:flux"]["models"].split(",")
|
||||
else:
|
||||
available_model_list = user_config["comfyui"]["models"].split(",")
|
||||
|
||||
valid_models = list(
|
||||
set(get_available_models())
|
||||
& set(user_config["comfyui"]["models"].split(","))
|
||||
set(get_available_models()) & set(available_model_list)
|
||||
)
|
||||
|
||||
if not valid_models:
|
||||
raise Exception("No valid models available.")
|
||||
|
||||
model = random.choice(valid_models)
|
||||
wf.set_node_param(model_node, model_param, model)
|
||||
|
||||
|
||||
# Generate image
|
||||
logging.debug(f"Generating image: {file_name}")
|
||||
results = api.queue_and_wait_images(wf, save_node)
|
||||
@ -244,8 +251,8 @@ def create_image(prompt: str | None = None) -> None:
|
||||
seed_param="seed",
|
||||
save_node="CivitAI Image Saver",
|
||||
save_param="filename",
|
||||
model_node=None, # FLUX doesn't use model selection
|
||||
model_param=None,
|
||||
model_node="CivitAI Image Saver",
|
||||
model_param="modelname",
|
||||
)
|
||||
else:
|
||||
generate_image("image", prompt)
|
||||
|
22
publish.sh
Normal file
22
publish.sh
Normal file
@ -0,0 +1,22 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Set variables
|
||||
IMAGE_NAME="ai-frame-image-server"
|
||||
REGISTRY="kithub.k-world.me.uk"
|
||||
USERNAME="karl"
|
||||
TAG="latest"
|
||||
FULL_IMAGE="$REGISTRY/$USERNAME/$IMAGE_NAME:$TAG"
|
||||
|
||||
# Build the image
|
||||
echo "🛠️ Building Docker image..."
|
||||
docker build -t $IMAGE_NAME .
|
||||
|
||||
# Tag the image
|
||||
echo "🏷️ Tagging image as $FULL_IMAGE"
|
||||
docker tag $IMAGE_NAME $FULL_IMAGE
|
||||
|
||||
# Push the image
|
||||
echo "📤 Pushing $FULL_IMAGE to $REGISTRY..."
|
||||
docker push $FULL_IMAGE
|
||||
|
||||
echo "✅ Done!"
|
@ -13,6 +13,9 @@ width = 1568
|
||||
height = 672
|
||||
FLUX = False
|
||||
|
||||
[comfyui:flux]
|
||||
models = flux1-dev-Q4_0.gguf,flux1-schnell-Q4_0.gguf
|
||||
|
||||
[openwebui]
|
||||
base_url = https://openwebui
|
||||
api_key = sk-
|
||||
|
Loading…
x
Reference in New Issue
Block a user