HaileyStorm
/

FLUX.1-Merges

Model card Files Files and versions Community

HaileyStorm commited on Aug 8

Commit

4385f55

•

1 Parent(s): b56725c

Update infer.py

Files changed (1) hide show

infer.py +2 -2

infer.py CHANGED Viewed

@@ -4,8 +4,8 @@ import os
 # Configuration
 MODEL_DIR = "./merged_models/10_1"
-IMAGE_OUTPUT_DIR = "./young"
-IMAGE_PREFIX = "ginny_10_1"
 DEVICE = torch.device("cpu")
 # If True, uses pipeline.enable_sequential_cpu_offload(). Make sure device is CPU.
 USE_CPU_OFFLOAD = True

 # Configuration
 MODEL_DIR = "./merged_models/10_1"
+IMAGE_OUTPUT_DIR = "./flowers"
+IMAGE_PREFIX = "flowers_10_1"
 DEVICE = torch.device("cpu")
 # If True, uses pipeline.enable_sequential_cpu_offload(). Make sure device is CPU.
 USE_CPU_OFFLOAD = True