diff --git a/workers/comfyui-json/client.py b/workers/comfyui-json/client.py index c877df2..93e184c 100644 --- a/workers/comfyui-json/client.py +++ b/workers/comfyui-json/client.py @@ -1,11 +1,11 @@ -from vastai import Serverless from .data_types import count_workload - import uuid import random import asyncio import random +from vastai import Serverless + async def main(): async with Serverless() as client: endpoint = await client.get_endpoint(name="my-comfy-endpoint") # Change this to your endpoint name diff --git a/workers/openai/client.py b/workers/openai/client.py index 1dadc68..8c88444 100644 --- a/workers/openai/client.py +++ b/workers/openai/client.py @@ -1,4 +1,3 @@ -#!/usr/bin/env python3 import logging import json import os