diff --git a/codes/hf.py b/codes/hf.py index 4f76ea8d534a4b0cec770e7f74e682748e535dd0..713345ceb13d7f47aa1714cb3270e43d368921f5 100644 --- a/codes/hf.py +++ b/codes/hf.py @@ -273,11 +273,13 @@ def find_groundstate_ham( """ model.nk=nk model.filling=filling - vectors = utils.generate_vectors(cutoff_Vk, model.dim) - model.vectors=[*model.int_model.keys()] + if model.int_model is not None: + model.vectors=[*model.int_model.keys()] + else: + model.vectors = utils.generate_vectors(cutoff_Vk, model.dim) if model.guess is None: model.random_guess(model.vectors) solver(model, optimizer, optimizer_kwargs) - model.vectors=[*model.int_model, *model.tb_model.keys()] + model.vectors=[*model.vectors, *model.tb_model.keys()] assert np.allclose((model.mf_k - np.moveaxis(model.mf_k, -1, -2).conj())/2, 0, atol=1e-15) return utils.hk2tb_model(model.hamiltonians_0 + model.mf_k, model.vectors, model.ks)