manbeast3b commited on
Commit
186709b
·
verified ·
1 Parent(s): 8b1080e

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -372,7 +372,7 @@ class BitNetQuantLayer(nn.Module):
372
 
373
  # Optimized SDXL Pipeline
374
  class OptimizedSDXLPipeline():
375
- def __init__(self, base_model_path: str = "stabilityai/stable-diffusion-xl-base-1.0"):
376
  # Load base pipeline
377
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
378
  self.pipeline = StableDiffusionXLPipeline.from_pretrained(base_model_path, torch_dtype=torch.float16).to(self.device)
 
372
 
373
  # Optimized SDXL Pipeline
374
  class OptimizedSDXLPipeline():
375
+ def __init__(self, base_model_path: str = "stablediffusionapi/newdream-sdxl-20"):
376
  # Load base pipeline
377
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
378
  self.pipeline = StableDiffusionXLPipeline.from_pretrained(base_model_path, torch_dtype=torch.float16).to(self.device)