Update src/pipeline.py
Browse files- 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 = "
|
| 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)
|