dreambooth_riffusion_model_currulao_v1/config.json

{"name": "model_1", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 64, 64, 4], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_5"}, "name": "input_5", "inbound_nodes": []}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 320], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_4"}, "name": "input_4", "inbound_nodes": []}, {"class_name": "InputLayer", "config": {"batch_input_shape": [null, 77, 768], "dtype": "float32", "sparse": false, "ragged": false, "name": "input_3"}, "name": "input_3", "inbound_nodes": []}, {"class_name": "UNetModel", "config": {}, "name": "u_net_model", "inbound_nodes": [[["input_5", 0, 0, {}], ["input_4", 0, 0, {}], ["input_3", 0, 0, {}]]]}], "input_layers": [["input_5", 0, 0], ["input_4", 0, 0], ["input_3", 0, 0]], "output_layers": [["u_net_model", 0, 0]]}