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"
|