Fix for windows
This commit is contained in:
@@ -259,9 +259,9 @@ class VibeVoiceForConditionalGeneration(VibeVoicePreTrainedModel):
|
|||||||
"constant",
|
"constant",
|
||||||
0,
|
0,
|
||||||
)
|
)
|
||||||
print("✅ Tied input and output embeddings using standard assignment.")
|
print("Tied input and output embeddings using standard assignment.")
|
||||||
else:
|
else:
|
||||||
print("ℹ️ tie_word_embeddings is False, not tying weights.")
|
print("tie_word_embeddings is False, not tying weights.")
|
||||||
|
|
||||||
# Also, ensure set_output_embeddings is safe, though your implementation looks okay.
|
# Also, ensure set_output_embeddings is safe, though your implementation looks okay.
|
||||||
# The key is to avoid calling it after accelerator.prepare().
|
# The key is to avoid calling it after accelerator.prepare().
|
||||||
|
|||||||
Reference in New Issue
Block a user