Update README.md

README.md

@@ -60,24 +60,38 @@ config = AutoConfig.from_pretrained(
     conv_dropout=0.1,
     skip_connections=True,
     conv_preprocessing_layers=[
-        {
-            "in_channels": 80,
-            "out_channels": 384,
-            "kernel_size": 5,
-            "padding": 2,
-            "activation": "gelu"
-        },
-        {
-            "in_channels": 384,
-            "out_channels": 384,
-            "kernel_size": 3,
-            "stride": 2,
-            "padding": 1,
-            "activation": "gelu"
-        }
-    ]
+          {
+              "in_channels": 128,
+              "out_channels": 1280,
+              "kernel_size": 3,
+              "stride": 1,
+              "padding": 1,
+              "activation": "gelu",
+              "bias": True
+          },
+          {
+              "in_channels": 1280,
+              "out_channels": 1280,
+              "kernel_size": 3,
+              "stride": 2,
+              "padding": 1,
+              "activation": "gelu",
+              "bias": True
+          }
+      ]
 )
 
+model = AutoModel.from_pretrained(
+    pretrained_model_name_or_path="mrprimenotes/sign-whisper-german",
+    config=config,
+    use_safetensors=True,
+    trust_remote_code=True,
+    ignore_mismatched_sizes=True,
+    torch_dtype=torch_dtype,
+    low_cpu_mem_usage=True,
+    device_map='auto'
+).to(device)
+
 tokenizer = AutoTokenizer.from_pretrained("mrprimenotes/sign-whisper-german")
 
 # raw model outputs: