# config.yaml
# Configuração central para a aplicação Deformes4D e seus especialistas.

application:
  workspace_dir: "deformes_workspace"

# Configuração para Hugging Face Spaces
sdk: gradio
app_file: app.py

specialists:

  vae:
    gpus_required: 2 # <-- DEDIQUE UMA GPU PARA O VAE

  llm_engine:
    provider: "gemini" # Opções: "llama", "gemini", etc.

  seedvr:
    gpus_required: 2

  llama_multimodal:
    gpus_required: 0 # <-- GARANTIR QUE ISTO É > 0
    model_id: "meta-llama/Llama-3.2-11B-Vision-Instruct"
    max_new_tokens: 4096
    temperature: 0.5
    top_p: 0.6
    
  flux:
    # Define quantas GPUs o pool do Flux deve tentar alocar.
    # Se não houver GPUs suficientes, o hardware_manager lançará um erro.
    # Se 0, usará a CPU.
    gpus_required: 0
  
  ltx:
    # Define quantas GPUs o pool do LTX deve tentar alocar.
    gpus_required: 4
    
    # Aponta para o arquivo de configuração específico do modelo LTX.
    # Alterado para usar o modelo 0.9.8-dev.
    config_file: "ltxv-13b-0.9.8-distilled-fp8.yaml"
    enable_prompt_enhancement: false
    lora:
      # Exemplo de como adicionar LoRAs para estilização.
      #- model_id: "Lightricks/LTX-Video-ICLoRA-pose-13b-0.9.7"
      #  weight: 0.5
      #- model_id: "Lightricks/LTX-Video-ICLoRA-depth-13b-0.9.7"
      #  weight: 0.3
      #- model_id: "Lightricks/LTX-Video-ICLoRA-detailer-13b-0.9.8"
      #  weight: 0.6
    
  vae_wan: # O VAE dedicado e treinado especificamente para o Wan2.2.
    gpus_required: 0 # Aloca 1 GPU para manter o VAE do Wan "quente".
    model_id: "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
  
  wan:
    gpus_required: 0 # Aloca 1 GPU para o worker do WanManager.
    model_id: "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
    main_model_id: "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
    optimized_transformer_id: "cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers"
    lora_repo: "Kijai/WanVideo_comfy"
    lora_filename: "Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors"
    # Parâmetros padrão para o modo Lightning (8 passos).
    default_height: 480
    default_width: 832
    default_frames: 33
    guidance_scale: 1.0
    guidance_scale_2: 1.0
    inference_steps: 4

  mmaudio:
    gpus_required: 2