Add free_dimension_overrides in config.json

#62
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -31,6 +31,14 @@
31
  "resi_connection": "1conv",
32
  "torch_dtype": "float32",
33
  "transformers_version": "4.34.0.dev0",
 
 
 
 
 
 
 
 
34
  "upsampler": "pixelshuffle",
35
  "upscale": 2,
36
  "use_absolute_embeddings": false,
 
31
  "resi_connection": "1conv",
32
  "torch_dtype": "float32",
33
  "transformers_version": "4.34.0.dev0",
34
+ "transformers.js_config": {
35
+ "free_dimension_overrides": {
36
+ "batch_size": 1,
37
+ "num_channels": 3,
38
+ "height": 16,
39
+ "width": 32
40
+ }
41
+ },
42
  "upsampler": "pixelshuffle",
43
  "upscale": 2,
44
  "use_absolute_embeddings": false,