Spaces:
Sleeping
Sleeping
add model link
Browse files- gradio_app.py +31 -1
gradio_app.py
CHANGED
|
@@ -27,6 +27,36 @@ CLIP_L_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffus
|
|
| 27 |
T5XXL_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffusion-3-medium/text_encoders/t5xxl_fp16.safetensors"
|
| 28 |
AE_PATH="/tiamat-vePFS/share_data/storage/huggingface/models/black-forest-labs/FLUX.1-dev/ae.safetensors"
|
| 29 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
# Load model function
|
| 31 |
def load_target_model():
|
| 32 |
logger.info("Loading models...")
|
|
@@ -201,7 +231,7 @@ with gr.Blocks() as demo:
|
|
| 201 |
sample_image = gr.Image(label="Upload a Conditional Image", type="pil")
|
| 202 |
|
| 203 |
# Frame number selection
|
| 204 |
-
frame_num = gr.Radio([4, 9], label="Select Frame Number", value=
|
| 205 |
|
| 206 |
# Seed and randomize seed options
|
| 207 |
seed = gr.Slider(0, np.iinfo(np.int32).max, step=1, label="Seed", value=0)
|
|
|
|
| 27 |
T5XXL_PATH="/tiamat-NAS/hailong/storage_backup/models/stabilityai/stable-diffusion-3-medium/text_encoders/t5xxl_fp16.safetensors"
|
| 28 |
AE_PATH="/tiamat-vePFS/share_data/storage/huggingface/models/black-forest-labs/FLUX.1-dev/ae.safetensors"
|
| 29 |
|
| 30 |
+
from huggingface_hub import hf_hub_download
|
| 31 |
+
repo_id = "Kijai/flux-fp8"
|
| 32 |
+
file_name = "flux1-dev-fp8.safetensors"
|
| 33 |
+
BASE_FLUX_CHECKPOINT = hf_hub_download(repo_id=repo_id, filename=file_name)
|
| 34 |
+
print(f"Downloaded BASE_FLUX_CHECKPOINT saved at: {file_path}")
|
| 35 |
+
|
| 36 |
+
from huggingface_hub import hf_hub_download
|
| 37 |
+
repo_id = "comfyanonymous/flux_text_encoders"
|
| 38 |
+
file_name = "clip_l.safetensors"
|
| 39 |
+
CLIP_L_PATH = hf_hub_download(repo_id=repo_id, filename=file_name)
|
| 40 |
+
print(f"Downloaded CLIP_L_PATH saved at: {file_path}")
|
| 41 |
+
|
| 42 |
+
from huggingface_hub import hf_hub_download
|
| 43 |
+
repo_id = "comfyanonymous/flux_text_encoders"
|
| 44 |
+
file_name = "t5xxl_fp8_e4m3fn.safetensors"
|
| 45 |
+
T5XXL_PATH = hf_hub_download(repo_id=repo_id, filename=file_name)
|
| 46 |
+
print(f"Downloaded T5XXL_PATH saved at: {file_path}")
|
| 47 |
+
|
| 48 |
+
from huggingface_hub import hf_hub_download
|
| 49 |
+
repo_id = "black-forest-labs/FLUX.1-dev"
|
| 50 |
+
file_name = "ae.safetensors"
|
| 51 |
+
AE_PATH = hf_hub_download(repo_id=repo_id, filename=file_name)
|
| 52 |
+
print(f"Downloaded AE_PATH saved at: {file_path}")
|
| 53 |
+
|
| 54 |
+
from huggingface_hub import hf_hub_download
|
| 55 |
+
repo_id = "showlab/makeanything"
|
| 56 |
+
file_name = "recraft_9f_lego.safetensors"
|
| 57 |
+
LORA_WEIGHTS_PATH = hf_hub_download(repo_id=repo_id, filename=file_name)
|
| 58 |
+
print(f"Downloaded LORA_WEIGHTS_PATH saved at: {file_path}")
|
| 59 |
+
|
| 60 |
# Load model function
|
| 61 |
def load_target_model():
|
| 62 |
logger.info("Loading models...")
|
|
|
|
| 231 |
sample_image = gr.Image(label="Upload a Conditional Image", type="pil")
|
| 232 |
|
| 233 |
# Frame number selection
|
| 234 |
+
frame_num = gr.Radio([4, 9], label="Select Frame Number", value=9)
|
| 235 |
|
| 236 |
# Seed and randomize seed options
|
| 237 |
seed = gr.Slider(0, np.iinfo(np.int32).max, step=1, label="Seed", value=0)
|