Update app.py
Browse files
app.py
CHANGED
|
@@ -22,9 +22,11 @@ from concurrent.futures import ThreadPoolExecutor
|
|
| 22 |
import uuid
|
| 23 |
|
| 24 |
DESCRIPTION = '''# Latent Consistency Model OpenVino CPU
|
| 25 |
-
Based on [Latency Consistency Model OpenVINO CPU](deinferno/Latent_Consistency_Model_OpenVino_CPU) HF space
|
| 26 |
|
| 27 |
-
Converted from [SoteMix](https://huggingface.co/Disty0/SoteMix) [
|
|
|
|
|
|
|
| 28 |
|
| 29 |
<p>Running on CPU 🥶.</p>
|
| 30 |
'''
|
|
@@ -33,7 +35,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
| 33 |
CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES") == "1"
|
| 34 |
|
| 35 |
model_id = "Disty0/LCM_SoteMix"
|
| 36 |
-
batch_size = 1
|
| 37 |
width = int(os.getenv("IMAGE_WIDTH", "512"))
|
| 38 |
height = int(os.getenv("IMAGE_HEIGHT", "512"))
|
| 39 |
num_images = int(os.getenv("NUM_IMAGES", "1"))
|
|
|
|
| 22 |
import uuid
|
| 23 |
|
| 24 |
DESCRIPTION = '''# Latent Consistency Model OpenVino CPU
|
| 25 |
+
Based on [Latency Consistency Model OpenVINO CPU](https://huggingface.co/deinferno/Latent_Consistency_Model_OpenVino_CPU) HF space
|
| 26 |
|
| 27 |
+
Converted from [SoteMix](https://huggingface.co/Disty0/SoteMix) to [LCM_SoteMix](https://huggingface.co/Disty0/LCM_SoteMix)
|
| 28 |
+
|
| 29 |
+
[Project page](https://latent-consistency-models.github.io)
|
| 30 |
|
| 31 |
<p>Running on CPU 🥶.</p>
|
| 32 |
'''
|
|
|
|
| 35 |
CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES") == "1"
|
| 36 |
|
| 37 |
model_id = "Disty0/LCM_SoteMix"
|
| 38 |
+
batch_size = -1
|
| 39 |
width = int(os.getenv("IMAGE_WIDTH", "512"))
|
| 40 |
height = int(os.getenv("IMAGE_HEIGHT", "512"))
|
| 41 |
num_images = int(os.getenv("NUM_IMAGES", "1"))
|