HaileyStorm
commited on
Update infer.py
Browse files
infer.py
CHANGED
@@ -4,8 +4,8 @@ import os
|
|
4 |
|
5 |
# Configuration
|
6 |
MODEL_DIR = "./merged_models/10_1"
|
7 |
-
IMAGE_OUTPUT_DIR = "./
|
8 |
-
IMAGE_PREFIX = "
|
9 |
DEVICE = torch.device("cpu")
|
10 |
# If True, uses pipeline.enable_sequential_cpu_offload(). Make sure device is CPU.
|
11 |
USE_CPU_OFFLOAD = True
|
|
|
4 |
|
5 |
# Configuration
|
6 |
MODEL_DIR = "./merged_models/10_1"
|
7 |
+
IMAGE_OUTPUT_DIR = "./flowers"
|
8 |
+
IMAGE_PREFIX = "flowers_10_1"
|
9 |
DEVICE = torch.device("cpu")
|
10 |
# If True, uses pipeline.enable_sequential_cpu_offload(). Make sure device is CPU.
|
11 |
USE_CPU_OFFLOAD = True
|