jhansss commited on
Commit
fa3bbac
·
1 Parent(s): f474b58

fix bilingual response following prompt updates

Browse files
Files changed (1) hide show
  1. modules/melody.py +3 -2
modules/melody.py CHANGED
@@ -25,7 +25,7 @@ class MelodyController:
25
  def get_melody_constraints(self, max_num_phrases: int = 5) -> str:
26
  """Return a lyric-format prompt based on melody structure."""
27
  if self.mode == "gen":
28
- return ""
29
 
30
  elif self.mode == "sample":
31
  assert self.database is not None, "Song database is not loaded."
@@ -46,10 +46,11 @@ class MelodyController:
46
  )
47
  + "\n如果没有足够的信息回答,请使用最少的句子,不要重复、不要扩展、不要加入无关内容。\n"
48
  )
49
- return prompt
50
 
51
  else:
52
  raise ValueError(f"Unsupported melody mode: {self.mode}")
 
 
53
 
54
  def generate_score(
55
  self, lyrics: str, language: str
 
25
  def get_melody_constraints(self, max_num_phrases: int = 5) -> str:
26
  """Return a lyric-format prompt based on melody structure."""
27
  if self.mode == "gen":
28
+ prompt = ""
29
 
30
  elif self.mode == "sample":
31
  assert self.database is not None, "Song database is not loaded."
 
46
  )
47
  + "\n如果没有足够的信息回答,请使用最少的句子,不要重复、不要扩展、不要加入无关内容。\n"
48
  )
 
49
 
50
  else:
51
  raise ValueError(f"Unsupported melody mode: {self.mode}")
52
+ prompt += "请使用用户输入的语言回答"
53
+ return prompt
54
 
55
  def generate_score(
56
  self, lyrics: str, language: str