ChuxiJ commited on
Commit
2eced9f
·
1 Parent(s): f4d9d31

fix create sample bug

Browse files
Files changed (1) hide show
  1. acestep/llm_inference.py +4 -4
acestep/llm_inference.py CHANGED
@@ -1554,7 +1554,7 @@ class LLMHandler:
1554
  if vocal_language and vocal_language.strip() and vocal_language.strip().lower() != "unknown":
1555
  # Use the specified language for constrained decoding
1556
  user_metadata = {"language": vocal_language.strip()}
1557
- skip_language = True # Skip language generation since we're injecting it
1558
  logger.info(f"Using user-specified language: {vocal_language.strip()}")
1559
 
1560
  # Generate using constrained decoding (inspiration phase)
@@ -1570,7 +1570,7 @@ class LLMHandler:
1570
  "target_duration": None, # No duration constraint
1571
  "user_metadata": user_metadata, # Inject language if specified
1572
  "skip_caption": False, # Generate caption
1573
- "skip_language": skip_language, # Skip if we're injecting language
1574
  "skip_genres": False, # Generate genres
1575
  "generation_phase": "understand", # Use understand phase for metadata + free-form lyrics
1576
  "caption": "",
@@ -1598,12 +1598,12 @@ class LLMHandler:
1598
  # Echo back the instrumental flag
1599
  metadata['instrumental'] = instrumental
1600
 
1601
- logger.info(f"Sample created successfully. Generated {len(metadata)} fields")
1602
  if constrained_decoding_debug:
1603
  logger.debug(f"Generated metadata: {list(metadata.keys())}")
1604
  logger.debug(f"Output text preview: {output_text[:300]}...")
1605
 
1606
- status_msg = f"✅ Sample created successfully\nGenerated fields: {', '.join(metadata.keys())}"
1607
  return metadata, status_msg
1608
 
1609
  def build_formatted_prompt_for_format(
 
1554
  if vocal_language and vocal_language.strip() and vocal_language.strip().lower() != "unknown":
1555
  # Use the specified language for constrained decoding
1556
  user_metadata = {"language": vocal_language.strip()}
1557
+ # skip_language = True # Skip language generation since we're injecting it
1558
  logger.info(f"Using user-specified language: {vocal_language.strip()}")
1559
 
1560
  # Generate using constrained decoding (inspiration phase)
 
1570
  "target_duration": None, # No duration constraint
1571
  "user_metadata": user_metadata, # Inject language if specified
1572
  "skip_caption": False, # Generate caption
1573
+ "skip_language": False,
1574
  "skip_genres": False, # Generate genres
1575
  "generation_phase": "understand", # Use understand phase for metadata + free-form lyrics
1576
  "caption": "",
 
1598
  # Echo back the instrumental flag
1599
  metadata['instrumental'] = instrumental
1600
 
1601
+ logger.info(f"Sample created successfully. Generated {metadata} fields")
1602
  if constrained_decoding_debug:
1603
  logger.debug(f"Generated metadata: {list(metadata.keys())}")
1604
  logger.debug(f"Output text preview: {output_text[:300]}...")
1605
 
1606
+ status_msg = f"✅ Sample created successfully\nGenerated fields: {metadata}"
1607
  return metadata, status_msg
1608
 
1609
  def build_formatted_prompt_for_format(