Update README.md

README.md

@@ -18,7 +18,7 @@ from flax.training import checkpoints
 flax.config.update('flax_use_orbax_checkpointing', False)
 # Load the model from HuggingFace
 from transformers import FlaxAutoModel
-wf = FlaxAutoModel.from_pretrained("rrende/j1j2_square_10x10", trust_remote_code=True)
+wf = FlaxAutoModel.from_pretrained("nqs-models/j1j2_square_10x10", trust_remote_code=True)
 N_params = nk.jax.tree_size(wf.params)
 print('Number of parameters = ', N_params, flush=True)
 lattice = nk.graph.Hypercube(length=10, n_dim=2, pbc=True, max_neighbor_order=2)
@@ -43,7 +43,7 @@ vstate = nk.vqs.MCState(sampler=sampler,
                         chunk_size=16384)
 # Overwrite samples with already thermalized ones
 from huggingface_hub import hf_hub_download
-path = hf_hub_download(repo_id="rrende/j1j2_square_10x10", filename="spins")
+path = hf_hub_download(repo_id="nqs-models/j1j2_square_10x10", filename="spins")
 samples = checkpoints.restore_checkpoint(ckpt_dir=path, prefix="spins", target=None)
 samples = jnp.array(samples, dtype='int8')
 vstate.sampler_state = vstate.sampler_state.replace(σ = samples)