llapsus commited on
Commit
4b873f9
·
verified ·
1 Parent(s): 87daef1

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -51,7 +51,7 @@ from safetensors.torch import load_file
51
  from galaxy_diffusion.models.vae import VAE
52
  from galaxy_diffusion.models.unet import LatentUNetCA
53
 
54
- path = snapshot_download("LLapsus/galaxy-diffusion") # downloads all files
55
  cfg = json.load(open(f"{path}/latent_diffusion_galaxy10_xattn_v1.config.json"))
56
 
57
  vae = VAE(**cfg["vae_config"])
 
51
  from galaxy_diffusion.models.vae import VAE
52
  from galaxy_diffusion.models.unet import LatentUNetCA
53
 
54
+ path = snapshot_download("llapsus/galaxy-diffusion") # downloads all files
55
  cfg = json.load(open(f"{path}/latent_diffusion_galaxy10_xattn_v1.config.json"))
56
 
57
  vae = VAE(**cfg["vae_config"])