akoksal commited on
Commit
7d6fae3
1 Parent(s): b3325e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)