Update app.py
Browse files
app.py
CHANGED
@@ -24,14 +24,14 @@ def load_pipeline(model_id):
|
|
24 |
NUM_CPU_CORES = psutil.cpu_count(logical=True)
|
25 |
|
26 |
# Cap the number of threads to the available physical cores
|
27 |
-
MAX_THREADS =
|
28 |
|
29 |
# Device and hardware configuration
|
30 |
DEVICE = "cpu"
|
31 |
|
32 |
# Model Options (optimized for CPU and memory constraints)
|
33 |
MODEL_OPTIONS = {
|
34 |
-
"Модель штучного інтелекту середня (Більше часу на виконання-краща якість)": "
|
35 |
#"Модель штучного інтелекту мала (Швидше-середня якість)": "hf-internal-testing/tiny-stable-diffusion-pipe",
|
36 |
}
|
37 |
|
|
|
24 |
NUM_CPU_CORES = psutil.cpu_count(logical=True)
|
25 |
|
26 |
# Cap the number of threads to the available physical cores
|
27 |
+
MAX_THREADS = min(8, NUM_CPU_CORES)
|
28 |
|
29 |
# Device and hardware configuration
|
30 |
DEVICE = "cpu"
|
31 |
|
32 |
# Model Options (optimized for CPU and memory constraints)
|
33 |
MODEL_OPTIONS = {
|
34 |
+
"Модель штучного інтелекту середня (Більше часу на виконання-краща якість)": "stabilityai/stable-diffusion-3-medium-diffusers",
|
35 |
#"Модель штучного інтелекту мала (Швидше-середня якість)": "hf-internal-testing/tiny-stable-diffusion-pipe",
|
36 |
}
|
37 |
|