Upload README.md with huggingface_hub
Browse files
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("
|
| 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"])
|