SSD float16 repo.
Browse files- run_ctrlx.py +1 -1
run_ctrlx.py
CHANGED
@@ -88,7 +88,7 @@ def inference(
|
|
88 |
def main(args):
|
89 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
90 |
|
91 |
-
model_id_or_path = "/
|
92 |
# refiner_id_or_path = "stabilityai/stable-diffusion-xl-refiner-1.0"
|
93 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
94 |
variant = "fp16" if device == "cuda" else "fp32"
|
|
|
88 |
def main(args):
|
89 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
90 |
|
91 |
+
model_id_or_path = "OzzyGT/SSD-1B"
|
92 |
# refiner_id_or_path = "stabilityai/stable-diffusion-xl-refiner-1.0"
|
93 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
94 |
variant = "fp16" if device == "cuda" else "fp32"
|