Upload unet/config.json with huggingface_hub

unet/config.json

@@ -2,10 +2,7 @@
   "_class_name": "UNet2DConditionModel",
   "_commit_hash": null,
   "_diffusers_version": "0.24.0",
-  "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--SimianLuo--LCM_Dreamshaper_v7/snapshots/4721097975058205c4edcdece2cc574b7dd7bc04/unet",
-  "_use_default_values": [
-    "reverse_transformer_layers_per_block"
-  ],
+  "_name_or_path": "/home/ubuntu/realtime-lcm-service/dreamshaper-8-lcm/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
@@ -44,7 +41,7 @@
   "mid_block_scale_factor": 1,
   "mid_block_type": "UNetMidBlock2DCrossAttn",
   "neuron": {
-    "auto_cast": "matmul",
+    "auto_cast": "all",
     "auto_cast_type": "bf16",
     "compiler_type": "neuronx-cc",
     "compiler_version": "2.11.0.35+4f5279863",
@@ -54,8 +51,7 @@
     "input_names": [
       "sample",
       "timestep",
-      "encoder_hidden_states",
-      "timestep_cond"
+      "encoder_hidden_states"
     ],
     "model_type": "unet",
     "output_attentions": false,
@@ -81,9 +77,9 @@
   "resnet_skip_time_act": false,
   "resnet_time_scale_shift": "default",
   "reverse_transformer_layers_per_block": null,
-  "sample_size": 96,
+  "sample_size": 64,
   "task": "semantic-segmentation",
-  "time_cond_proj_dim": 256,
+  "time_cond_proj_dim": null,
   "time_embedding_act_fn": null,
   "time_embedding_dim": null,
   "time_embedding_type": "positional",