Spaces:
Running
on
A100
Running
on
A100
fix create sample bug
Browse files- 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":
|
| 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 {
|
| 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: {
|
| 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(
|