add umt5 support

This commit is contained in:
leejet committed 2025-08-09 16:07:04 +08:00
1 parent 5f7d98884c
commit bace0a08c4
12 files changed
+762430 -81

No files matched your search

+5 -5
View File
@@ -27,7 +27,7 @@
#define SAFE_STR(s) ((s) ? (s) : "")
#define BOOL_STR(b) ((b) ? "true" : "false")
#include "wan.hpp"
#include "t5.hpp"
const char* modes_str[] = {
"img_gen",
@@ -746,11 +746,11 @@ void sd_log_cb(enum sd_log_level_t level, const char* log, void* data) {
int main(int argc, const char* argv[]) {
SDParams params;
params.verbose = true;
sd_set_log_callback(sd_log_cb, (void*)&params);
// params.verbose = true;
// sd_set_log_callback(sd_log_cb, (void*)&params);
WAN::WanRunner::load_from_file_and_test(argv[1]);
return 0;
// T5Embedder::load_from_file_and_test(argv[1]);
// return 0;
parse_args(argc, argv, params);