bennykok
018d9a7b8d
For self hosting instance, this quick fix enable you to have the civitai token appended to the download url. Add CIVITAI_TOKEN in Fly io
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
import json
|
|
import requests
|
|
import time
|
|
import subprocess
|
|
|
|
print("Starting ComfyUI")
|
|
|
|
command = ["python", "main.py", "--disable-auto-launch", "--disable-metadata", "--cpu"]
|
|
# Start the server
|
|
server_process = subprocess.Popen(command, cwd="/comfyui")
|
|
|
|
def check_server(url, retries=50, delay=500):
|
|
for i in range(retries):
|
|
try:
|
|
response = requests.head(url)
|
|
|
|
# If the response status code is 200, the server is up and running
|
|
if response.status_code == 200:
|
|
print(f"builder - API is reachable")
|
|
return True
|
|
except requests.RequestException as e:
|
|
# If an exception occurs, the server may not be ready
|
|
pass
|
|
|
|
# Wait for the specified delay before retrying
|
|
time.sleep(delay / 1000)
|
|
|
|
print(
|
|
f"builder- Failed to connect to server at {url} after {retries} attempts."
|
|
)
|
|
return False
|
|
|
|
root_url = "http://127.0.0.1:8188"
|
|
|
|
check_server(root_url)
|
|
|
|
headers = {"Content-Type": "application/json"}
|
|
|
|
# Load JSON array from deps.json
|
|
with open('deps.json') as f:
|
|
packages = json.load(f)
|
|
|
|
# Make a POST request for each package
|
|
for package in packages:
|
|
response = requests.request("POST", f"{root_url}/customnode/install", json=package, headers=headers)
|
|
print(response.text)
|
|
|
|
with open('models.json') as f:
|
|
models = json.load(f)
|
|
|
|
for model in models:
|
|
import os
|
|
if "civitai.com/api" in model['url'] and not "token=" in model['url']:
|
|
if "?" in model['url']:
|
|
model['url'] += "&token=" + os.environ.get('CIVITAI_TOKEN', '')
|
|
else:
|
|
model['url'] += "?token=" + os.environ.get('CIVITAI_TOKEN', '')
|
|
response = requests.request("POST", f"{root_url}/model/install", json=model, headers=headers)
|
|
print(response.text)
|
|
|
|
# Close the server
|
|
server_process.terminate()
|
|
print("Finished installing dependencies.") |