shithub: opus

Info  •  Files  •  Log  •  Branches

Fix rdovae exporting

--- a/dnn/torch/weight-exchange/wexchange/c_export/common.py

+++ b/dnn/torch/weight-exchange/wexchange/c_export/common.py

@@ -40,6 +40,7 @@

     dtype_suffix = {

         'float' : 'float',

+        'opus_uint8' : 'uint8',

         'opus_int8' : 'int8',

         'opus_uint16' : 'uint16',

         'opus_int16' : 'int16',

@@ -54,7 +55,7 @@

 #ifndef USE_WEIGHTS_FILE

 '''

         )

-    writer.weight_arrays.append(name)

+        writer.weight_arrays.append(name)

     if reshape_8x4:

         vector = vector.reshape((vector.shape[0]//4, 4, vector.shape[1]//8, 8))

-- 

⑨