Update README.md
Browse files
README.md
CHANGED
@@ -75,7 +75,7 @@ longform_model = AutoModelForCausalLM.from_pretrained("akoksal/LongForm-LLaMA-7B
|
|
75 |
# Add diff with base model
|
76 |
for name, param in base_model.named_parameters():
|
77 |
longform_model.state_dict()[name].copy_(param + longform_model_diff.state_dict()[name])
|
78 |
-
|
79 |
# 4. Example
|
80 |
instruction = "Write an essay about meditation. [EOI]"
|
81 |
torch.manual_seed(42)
|
|
|
75 |
# Add diff with base model
|
76 |
for name, param in base_model.named_parameters():
|
77 |
longform_model.state_dict()[name].copy_(param + longform_model_diff.state_dict()[name])
|
78 |
+
del base_model, longform_model_diff
|
79 |
# 4. Example
|
80 |
instruction = "Write an essay about meditation. [EOI]"
|
81 |
torch.manual_seed(42)
|