thanks to Eagelaxis ❤

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+engine/unet.plan filter=lfs diff=lfs merge=lfs -text

engine.json

@@ -0,0 +1 @@
+{"trt_version": "8.6.0", "max_batch_size": 1, "refit": false, "static_batch": false, "dynamic_shape": true, "all_tactics": false, "preview_features": true, "optimize_height": 512, "optimize_width": 512, "min_latent_shape": 256, "max_latent_shape": 1024}

engine/unet.plan

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98d62fbb9be1f2636333b9e780188f9b67670ff8b1bc37256efbb22816ceb146
+size 1740279188

onnx/unet.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:69d49e8361f462ea78fc1e2c2c396286409580ebb27b0040ca68e8fb7d7f218c
+size 1721128005

onnx/unet.opt.onnx

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