Update inference_configs/inference.yaml

inference_configs/inference.yaml

@@ -1,15 +1,18 @@
+
 # Configuração do Modelo
 model_name: "flux-dev"
 ckpt_dir: ./pretrained_weights/ByteMorpher/dit.safetensors
 
 # --- Configurações de Condicionamento de Imagem ---
-# Esta é a seção mais importante para a edição de imagem para imagem.
-# O app.py corrigido agora utiliza estas configurações corretamente.
-use_share_weight_referencenet: true   # ATIVADO: Essencial para usar a imagem de entrada como referência.
-use_spatial_condition: false          # DESATIVADO: Geralmente, usa-se um ou outro método de condicionamento.
-share_position_embedding: true        # MANTIDO: Importante para o alinhamento espacial entre a entrada e a saída.
+use_share_weight_referencenet: true
+use_spatial_condition: false
+share_position_embedding: true
+
+# --- NOVO: Especifique qual parte da rede usar como referência ---
+double_block_refnet: true   # ATIVADO: Usa os blocos iniciais (double stream).
+single_block_refnet: false  # DESATIVADO: Não usa os blocos finais (single stream).
 
-# --- Configurações de IP-Adapter e LoRA (Desativados por padrão) ---
+# --- Configurações de IP-Adapter e LoRA (Desativados) ---
 use_ip: false
 ip_local_path: null
 ip_repo_id: null
@@ -18,19 +21,19 @@ ip_scale: 1.0
 use_lora: false
 rank: 16
 
-# --- Parâmetros de Amostragem (Configurações de Alta Qualidade) ---
-sample_width: 1024       # ALTA QUALIDADE: Resolução aumentada para 1024x1024.
-sample_height: 1024      # ALTA QUALIDADE: Resolução aumentada para 1024x1024.
-sample_steps: 50         # ALTA QUALIDADE: 50 passos é um ótimo equilíbrio entre velocidade e detalhe.
-cfg_scale: 4.0           # ALTA QUALIDADE: Aumentado para maior fidelidade ao prompt de texto.
-seed: -1                 # DINÂMICO: -1 para gerar uma imagem diferente a cada execução. Mude para um número (ex: 42) para resultados reprodutíveis.
-mixed_precision: "bf16"  # OTIMIZADO: bfloat16 é ideal e mais estável para GPUs L40.
+# --- Parâmetros de Amostragem (Alta Qualidade) ---
+sample_width: 1024
+sample_height: 1024
+sample_steps: 50
+cfg_scale: 4.0
+seed: -1
+mixed_precision: "bf16"
 
-# --- Configuração de Dados e Logging (geralmente não precisam de alteração para a demo) ---
+# --- Configuração de Dados e Logging ---
 data_config:
   eval_batch_size: 8
   num_workers: 4
-  img_size: 1024         # Atualizado para corresponder à resolução de amostragem.
+  img_size: 1024
   img_dir: output_bench/
   grayscale_editing: false
   zoom_camera: false