makewong commited on
Commit
045312c
1 Parent(s): eb2919a

Update pic2song.py

Browse files
Files changed (1) hide show
  1. pic2song.py +1 -1
pic2song.py CHANGED
@@ -70,7 +70,7 @@ def predict_step(image_paths):
70
  @torch.no_grad()
71
  @torch.inference_mode()
72
  def pic2song(image_paths, image_ins):
73
- in_max_new_tokens = int(image_ins * 30)
74
  # 调用predict_step函数并输出预测的图片描述文本列表
75
  print(image_paths)
76
  user_message = predict_step(image_paths)
 
70
  @torch.no_grad()
71
  @torch.inference_mode()
72
  def pic2song(image_paths, image_ins):
73
+ in_max_new_tokens = int(image_ins * 60)
74
  # 调用predict_step函数并输出预测的图片描述文本列表
75
  print(image_paths)
76
  user_message = predict_step(image_paths)