G-Rost commited on
Commit
6234b1e
·
verified ·
1 Parent(s): 9d06d3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = max(320, 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
- "Модель штучного інтелекту середня (Більше часу на виконання-краща якість)": "CompVis/stable-diffusion-v1-4",
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