xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-29 19:07:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean up

Browse Source
This commit is contained in:
beta9 2024-10-20 20:22:06 -04:00
parent 6ef45cb0ce
commit 3cb0220cd9
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
app.py
View File

@ -2,45 +2,45 @@ from beam import task_queue, Output, Image, env, Volume
import os import os
import sys import sys
import requests import requests
if env.is_remote(): if env.is_remote():
import folder_paths import folder_paths
sys.path.append(os.getcwd())
import main sys.path.append(os.getcwd())
import main
# Function to setup environment and load models # Function to setup environment and load models
def upload_models(): def upload_models():
# Define the model URLs and their respective directories # Define the model URLs and their respective directories
model_urls = [ model_urls = [
#put models here # put models here
] ]
# Download models if not already downloaded # Download models if not already downloaded
for model_name, filename, directory in model_urls: for model_name, filename, directory in model_urls:
if not os.path.exists(f"{directory}/{filename}"): if not os.path.exists(f"{directory}/{filename}"):
print(os.getcwd()) print(os.getcwd())
os.system(f"wget -c https://huggingface.co/{model_name}/resolve/main/{filename} -P {os.path.join(os.getcwd() , directory)}") os.system(
f"wget -c https://huggingface.co/{model_name}/resolve/main/{filename} -P {os.path.join(os.getcwd() , directory)}"
)
# Import functions or execute code from workflow_api.py # Import functions or execute code from workflow_api.py
# Step 1: Define environment setup and model loading # Step 1: Define environment setup and model loading
@endpoint( @task_queue(
name="ComfyUI", name="ComfyUI",
cpu=4, cpu=4,
memory="32Gi", memory="32Gi",
gpu="A10G", gpu="A10G",
image=Image( image=Image(
python_version="python3.10", python_version="python3.10",
python_packages=requirements.txt, python_packages="requirements.txt",
#commands=["pip3 install opencv-python"], # commands=["pip3 install opencv-python"],
), ),
volumes=[Volume(name="PUTVOLUMEHERE", mount_path="./VOLUMEPATH)], volumes=[Volume(name="PUTVOLUMEHERE", mount_path="./VOLUMEPATH")],
#keep_warm_seconds=0, # keep_warm_seconds=0,
#on_start=upload_models, # on_start=upload_models,
) )
def generate_video(**inputs): def generate_video(**inputs):
while(true): main()
main()