Skip to content

C
coderai
Commit 3c682962 authored by Your Name's avatar Your Name
Browse files
Options

Fix: Use StableDiffusion class for sd.cpp fallback

parent 5b818bc6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
coderai
View file @ 3c682962
......@@ -4597,10 +4597,10 @@ def main():
print(f"Trying stable-diffusion-cpp-python fallback...")
# Try stable-diffusion-cpp-python as fallback
try:
import stable_diffusion_cpp as sd
from stable_diffusion_cpp import StableDiffusion
print(f"Loading with sd.cpp: {model_path}")
sd_model = sd.load_model(model_path)
sd_model = StableDiffusion(model_path=model_path)
multi_model_manager.add_model(model_key, sd_model)
print(f"Image model loaded successfully via sd.cpp: {original_model_name}")
except ImportError as sd_error:
......@@ -5009,10 +5009,10 @@ def main():
print(f"Trying stable-diffusion-cpp-python fallback...")
# Try stable-diffusion-cpp-python as fallback
try:
import stable_diffusion_cpp as sd
from stable_diffusion_cpp import StableDiffusion
print(f"Loading with sd.cpp: {model_path}")
sd_model = sd.load_model(model_path)
sd_model = StableDiffusion(model_path=model_path)
multi_model_manager.add_model(model_key, sd_model)
print(f"Image model loaded successfully via sd.cpp: {original_model_name}")
except ImportError as sd_error:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment