Upgrade to 1.2

app.py

@@ -73,9 +73,9 @@ hf_token = os.environ.get("HF_TOKEN")
 lora_v1_path = snapshot_download(repo_id="mrfakename/EmoAct-MiMo", token=hf_token)
 print(f"LoRA v1.0 weights downloaded to: {lora_v1_path}")
 
-print("Downloading EmoAct-MiMo v1.1 (Beta) LoRA weights...")
-lora_v1_1_path = snapshot_download(repo_id="mrfakename/EmoAct-MiMo-v1.1", token=hf_token)
-print(f"LoRA v1.1 (Beta) weights downloaded to: {lora_v1_1_path}")
+print("Downloading EmoAct-MiMo v1.2 (Beta) LoRA weights...")
+lora_v1_1_path = snapshot_download(repo_id="mrfakename/EmoAct-MiMo-v1.2", token=hf_token)
+print(f"LoRA v1.2 (Beta) weights downloaded to: {lora_v1_1_path}")
 
 # Load tokenizer and get special tokens
 print("Loading tokenizer...")
@@ -123,30 +123,30 @@ merged_model_v1.save_pretrained(merged_model_v1_path)
 tokenizer.save_pretrained(merged_model_v1_path)
 print(f"Merged model v1.0 saved to {merged_model_v1_path}")
 
-# Load base model for v1.1
-print("Loading base MiMo-Audio model for v1.1...")
+# Load base model for v1.2
+print("Loading base MiMo-Audio model for v1.2...")
 base_model_v1_1 = MiMoAudioForCausalLM.from_pretrained(
     base_model_path,
     args=model_args,
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
-print("Base model v1.1 loaded")
+print("Base model v1.2 loaded")
 
-# Load and merge LoRA v1.1
-print("Loading LoRA v1.1 (Beta) adapter...")
+# Load and merge LoRA v1.2
+print("Loading LoRA v1.2 (Beta) adapter...")
 model_with_lora_v1_1 = PeftModel.from_pretrained(base_model_v1_1, lora_v1_1_path)
-print("Merging LoRA v1.1 (Beta) weights...")
+print("Merging LoRA v1.2 (Beta) weights...")
 merged_model_v1_1 = model_with_lora_v1_1.merge_and_unload()
-print("LoRA v1.1 (Beta) weights merged!")
+print("LoRA v1.2 (Beta) weights merged!")
 
-# Save merged model v1.1 to temporary directory
-print("Saving merged model v1.1...")
+# Save merged model v1.2 to temporary directory
+print("Saving merged model v1.2...")
 merged_model_v1_1_path = "/tmp/merged_mimo_audio_v1_1"
 os.makedirs(merged_model_v1_1_path, exist_ok=True)
 merged_model_v1_1.save_pretrained(merged_model_v1_1_path)
 tokenizer.save_pretrained(merged_model_v1_1_path)
-print(f"Merged model v1.1 (Beta) saved to {merged_model_v1_1_path}")
+print(f"Merged model v1.2 (Beta) saved to {merged_model_v1_1_path}")
 
 # Initialize both MimoAudio models
 print("Initializing MimoAudio wrappers...")
@@ -163,7 +163,7 @@ print("Both models ready!")
 # Dictionary to store models
 models = {
     "EmoAct-MiMo v1.0 (Stable)": model_v1,
-    "EmoAct-MiMo v1.1 (Beta - Experimental)": model_v1_1
+    "EmoAct-MiMo v1.2 (Beta - Experimental)": model_v1_1
 }
 
 @spaces.GPU
@@ -218,10 +218,10 @@ with gr.Blocks(title="EmoAct-MiMo TTS") as demo:
     with gr.Row():
         with gr.Column():
             model_selector = gr.Dropdown(
-                choices=["EmoAct-MiMo v1.0 (Stable)", "EmoAct-MiMo v1.1 (Beta - Experimental)"],
+                choices=["EmoAct-MiMo v1.0 (Stable)", "EmoAct-MiMo v1.2 (Beta - Experimental)"],
                 value="EmoAct-MiMo v1.0 (Stable)",
                 label="Model Selection",
-                info="v1.0 is the current stable model. v1.1 is a beta experimental version with potentially different characteristics."
+                info="v1.0 is the current stable model. v1.2 is a beta experimental version with potentially different characteristics."
             )
             emotion_input = gr.Textbox(
                 label="Emotion",
@@ -269,12 +269,12 @@ with gr.Blocks(title="EmoAct-MiMo TTS") as demo:
                 "YES! YES! I DID IT! (laughs breathlessly) I can't believe it actually worked! This is... this is everything I've ever dreamed of! I'm so happy I could cry!"
             ],
             [
-                "EmoAct-MiMo v1.1 (Beta - Experimental)",
+                "EmoAct-MiMo v1.2 (Beta - Experimental)",
                 "crushing despair, hopelessness, depression, and deep emotional pain, speaking without any accent",
                 "What's the point anymore? I've tried everything. Nothing changes. Nothing ever gets better. I'm so tired of pretending I'm okay when I'm falling apart inside."
             ],
             [
-                "EmoAct-MiMo v1.1 (Beta - Experimental)",
+                "EmoAct-MiMo v1.2 (Beta - Experimental)",
                 "bitter jealousy, envy, resentment, and seething frustration, speaking without any accent",
                 "Of course they chose you. They always choose you. <laugh> Must be nice, having everything handed to you while the rest of us break our backs. You don't even appreciate what you have."
             ]