fixed prediction selection logic

Author: daniandtheweb

examples/cli/main.cpp

@@ -49,6 +49,7 @@ const char* schedule_str[] = {
 };
 
 const char* prediction_str[] = {
+    "default",
     "eps",
     "v",
     "flow",
@@ -104,7 +105,7 @@ struct SDParams {
 
     sample_method_t sample_method = EULER_A;
     schedule_t schedule           = DEFAULT;
-    prediction_t prediction       = EPS_PRED;
+    prediction_t prediction       = DEFAULT_PRED;
     int sample_steps              = 20;
     float strength                = 0.75f;
     float control_strength        = 0.9f;

stable-diffusion.cpp

@@ -423,7 +423,7 @@ class StableDiffusionGGML {
         LOG_INFO("loading model from '%s' completed, taking %.2fs", model_path.c_str(), (t1 - t0) * 1.0f / 1000);
 
 
-        if (prediction != EPS_PRED) {
+        if (prediction != DEFAULT_PRED) {
             switch (prediction) {
                 case EPS_PRED:
                     LOG_INFO("running in eps-prediction mode");

stable-diffusion.h

@@ -54,6 +54,7 @@ enum schedule_t {
 };
 
 enum prediction_t {
+    DEFAULT_PRED,
     EPS_PRED,
     V_PRED,
     FLOW_PRED,