release code

model/model_manager.py

@@ -7,7 +7,7 @@ import spaces
 import torch
 from PIL import Image
 from openai import OpenAI
-from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, VIDEO_GENERATION_MODELS, load_pipeline
+from .models import IMAGE_GENERATION_MODELS, VIDEO_GENERATION_MODELS, load_pipeline
 from serve.upload import get_random_mscoco_prompt, get_random_video_prompt, get_ssh_random_video_prompt, get_ssh_random_image_prompt
 from serve.constants import SSH_CACHE_OPENSOURCE, SSH_CACHE_ADVANCE, SSH_CACHE_PIKA, SSH_CACHE_SORA, SSH_CACHE_IMAGE