Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -62,7 +62,7 @@ import torch # Ensure torch is imported if needed for device placement
|
|
| 62 |
# --- 1. Load Model ---
|
| 63 |
# Load the model from Hugging Face Hub
|
| 64 |
# Make sure you are logged in (`huggingface-cli login`) if the repo is private.
|
| 65 |
-
model_id = "
|
| 66 |
device = "cuda" if torch.cuda.is_available() else "cpu" # Example device placement
|
| 67 |
# Note: The KimiAudio class might handle model loading differently.
|
| 68 |
# You might need to pass the model_id directly or download checkpoints manually
|
|
|
|
| 62 |
# --- 1. Load Model ---
|
| 63 |
# Load the model from Hugging Face Hub
|
| 64 |
# Make sure you are logged in (`huggingface-cli login`) if the repo is private.
|
| 65 |
+
model_id = "moonshotai/Kimi-Audio-7B-Instruct" # Or "Kimi/Kimi-Audio-7B"
|
| 66 |
device = "cuda" if torch.cuda.is_available() else "cpu" # Example device placement
|
| 67 |
# Note: The KimiAudio class might handle model loading differently.
|
| 68 |
# You might need to pass the model_id directly or download checkpoints manually
|