alphard/stable-diffusion.cpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: print seed value when args.seed < 0

Browse Source
This commit is contained in:
leejet 2023-08-22 23:03:17 +08:00
parent 721cb324af
commit 0d7f04b135
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
examples/main.cpp
View File

@ -1,4 +1,5 @@
#include <stdio.h> #include <stdio.h>
#include <ctime>
#include <fstream> #include <fstream>
#include <iostream> #include <iostream>
#include <random> #include <random>
@ -28,18 +29,17 @@
#define TXT2IMG "txt2img" #define TXT2IMG "txt2img"
#define IMG2IMG "img2img" #define IMG2IMG "img2img"
// get_num_physical_cores is copy from // get_num_physical_cores is copy from
// https://github.com/ggerganov/llama.cpp/blob/master/examples/common.cpp // https://github.com/ggerganov/llama.cpp/blob/master/examples/common.cpp
// LICENSE: https://github.com/ggerganov/llama.cpp/blob/master/LICENSE // LICENSE: https://github.com/ggerganov/llama.cpp/blob/master/LICENSE
int32_t get_num_physical_cores() { int32_t get_num_physical_cores() {
#ifdef __linux__ #ifdef __linux__
// enumerate the set of thread siblings, num entries is num cores // enumerate the set of thread siblings, num entries is num cores
std::unordered_set<std::string> siblings; std::unordered_set<std::string> siblings;
for (uint32_t cpu=0; cpu < UINT32_MAX; ++cpu) { for (uint32_t cpu = 0; cpu < UINT32_MAX; ++cpu) {
std::ifstream thread_siblings("/sys/devices/system/cpu" std::ifstream thread_siblings("/sys/devices/system/cpu" + std::to_string(cpu) + "/topology/thread_siblings");
+ std::to_string(cpu) + "/topology/thread_siblings");
if (!thread_siblings.is_open()) { if (!thread_siblings.is_open()) {
break; // no more cpus break; // no more cpus
} }
std::string line; std::string line;
if (std::getline(thread_siblings, line)) { if (std::getline(thread_siblings, line)) {
@ -61,7 +61,7 @@ int32_t get_num_physical_cores() {
return num_physical_cores; return num_physical_cores;
} }
#elif defined(_WIN32) #elif defined(_WIN32)
//TODO: Implement // TODO: Implement
#endif #endif
unsigned int n_threads = std::thread::hardware_concurrency(); unsigned int n_threads = std::thread::hardware_concurrency();
return n_threads > 0 ? (n_threads <= 4 ? n_threads : n_threads / 2) : 4; return n_threads > 0 ? (n_threads <= 4 ? n_threads : n_threads / 2) : 4;
@ -282,6 +282,11 @@ void parse_args(int argc, const char* argv[], Option* opt) {
fprintf(stderr, "error: can only work with strength in [0.0, 1.0]\n"); fprintf(stderr, "error: can only work with strength in [0.0, 1.0]\n");
exit(1); exit(1);
} }
if (opt->seed < 0) {
srand((int)time(NULL));
opt->seed = rand();
}
} }
int main(int argc, const char* argv[]) { int main(int argc, const char* argv[]) {