Upload folder using huggingface_hub

config.json

@@ -0,0 +1,26 @@
+{
+  "image_size": 256,
+  "latent_dim": 16,
+  "ae_hidden_dim": 64,
+  "ae_num_blocks": 3,
+  "hidden_dim": 256,
+  "num_hidden_layers": 12,
+  "num_heads": 8,
+  "patch_size": 4,
+  "text_encoder_name": "./all-MiniLM-L6-v2",
+  "text_encoder_git_url": "https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2",
+  "text_encoder_dim": 384,
+  "num_timesteps": 1000,
+  "beta_start": 0.0001,
+  "beta_end": 0.02,
+  "learning_rate": 0.0001,
+  "vae_pretrain_epochs": 1,
+  "vae_pretrain_lr": 0.001,
+  "grad_clip_norm": 1.0,
+  "warmup_steps": 500,
+  "batch_size": 8,
+  "num_epochs": 5,
+  "model_name": "ShellD",
+  "_class_name": "ShellD",
+  "_version": "0.1"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d069394a6772b0da3b35e287db390772494dcef5b2baeba1db76949315580d0
+size 267847332