Update inference_configs/inference.yaml

inference_configs/inference.yaml

@@ -1,33 +1,44 @@
+# Configuração do Modelo
 model_name: "flux-dev"
-use_spatial_condition: false
-share_position_embedding: false
-use_share_weight_referencenet: true 
+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.
+
+# --- Configurações de IP-Adapter e LoRA (Desativados por padrão) ---
 use_ip: false
 ip_local_path: null
 ip_repo_id: null
 ip_name: null
 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.
+
+# --- Configuração de Dados e Logging (geralmente não precisam de alteração para a demo) ---
 data_config:
-  eval_batch_size: 0
+  eval_batch_size: 8
   num_workers: 4
-  img_size: 768
-  img_dir: output_bench/ #./ByteMorph-Bench/
+  img_size: 1024         # Atualizado para corresponder à resolução de amostragem.
+  img_dir: output_bench/
   grayscale_editing: false
   zoom_camera: false
   random_ratio: false
 report_to: wandb
 eval_batch_size: 1
-ckpt_dir: ./pretrained_weights/ByteMorpher/dit.safetensors
 output_dir: ./test_log/seedmorpher/
 logging_dir: logs
-mixed_precision: "fp16"
-rank: 16
-single_blocks: null
-double_blocks: null
 disable_sampling: false
-sample_width: 768
-sample_height: 768
-sample_steps: 23
-seed: 42
-cfg_scale: 2
+single_blocks: null
+double_blocks: null