Method close

train/utils.py:64–67

Close the JSON array and the file.

(self)

        self.file.flush()
    
    def close(self):
        """Close the JSON array and the file."""
        self.file.write('\n]')
        self.file.flush()


def get_base_model_state_dict_from_peft(peft_state_dict, lora_alpha, lora_r):