twodgirl commited on
Commit
6df6312
1 Parent(s): 5c31d1f

SSD float16 repo.

Browse files
Files changed (1) hide show
  1. 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 = "/mnt/newhome/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"
 
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"