Update README.md
Browse files
README.md
CHANGED
@@ -24,7 +24,7 @@ finetuned_embedding_layer = finetuned_model.get_input_embeddings()
|
|
24 |
with torch.no_grad(): # Very important: No gradient tracking during this operation!
|
25 |
finetuned_embedding_layer.weight.copy_(base_embedding_layer.weight)
|
26 |
|
27 |
-
print(finetuned_model.get_input_embeddings().weight.shape)
|
28 |
|
29 |
# --- 4. Save the Modified Base Model ---
|
30 |
output_dir = "QwQ-32B-Snowdrop-v0-EmbedFix"
|
|
|
24 |
with torch.no_grad(): # Very important: No gradient tracking during this operation!
|
25 |
finetuned_embedding_layer.weight.copy_(base_embedding_layer.weight)
|
26 |
|
27 |
+
print(finetuned_model.get_input_embeddings().weight.shape) # Verify this is the size we want it
|
28 |
|
29 |
# --- 4. Save the Modified Base Model ---
|
30 |
output_dir = "QwQ-32B-Snowdrop-v0-EmbedFix"
|