euiiiia commited on
Commit
9c9e6c2
·
verified ·
1 Parent(s): 3a74b5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -35,12 +35,26 @@ def run_generate_low(prompt, neg_prompt, start_img, height, width, duration, cfg
35
  """Executa a primeira etapa: geração de um vídeo base em baixa resolução."""
36
  print("UI: Chamando generate_low")
37
  if True:
 
38
  conditioning_items = []
39
  if start_img:
40
- num_frames_estimate = int(duration * 24)
41
- items_list = [[start_img, 0, 1.0]]
42
- conditioning_items = video_generation_service.prepare_condition_items(items_list, height, width, num_frames_estimate)
43
-
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  used_seed = None if randomize_seed else seed
45
 
46
  video_path, tensor_path, final_seed = video_generation_service.generate_low_resolution(
 
35
  """Executa a primeira etapa: geração de um vídeo base em baixa resolução."""
36
  print("UI: Chamando generate_low")
37
  if True:
38
+
39
  conditioning_items = []
40
  if start_img:
41
+ print(f"UI: Preparando latentes de condicionamento para a imagem: {start_img}")
42
+
43
+ image_tensor = load_image_to_tensor_with_resize_and_crop(start_img, height, width)
44
+ image_tensor = image_tensor.unsqueeze(0).unsqueeze(2)
45
+ with torch.no_grad():
46
+ conditioning_latents = vae_manager_singleton.encode(image_tensor)
47
+ print(f"UI: Latentes de condicionamento criados. Shape: {conditioning_latents.shape}")
48
+
49
+ # Criamos a lista de itens aqui e a passamos diretamente.
50
+ conditioning_items = [
51
+ PatchedConditioningItem(
52
+ latents=conditioning_latents,
53
+ media_frame_number=0,
54
+ conditioning_strength=1.0
55
+ )
56
+ ]
57
+
58
  used_seed = None if randomize_seed else seed
59
 
60
  video_path, tensor_path, final_seed = video_generation_service.generate_low_resolution(