Ashley Wright
commited on
Commit
•
7a19876
1
Parent(s):
697cabb
Load from folder instead of cached repository
Browse files
.gitmodules
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
[submodule "sdxl-newdream-20"]
|
| 2 |
-
path = models/
|
| 3 |
url = https://huggingface.co/stablediffusionapi/newdream-sdxl-20
|
| 4 |
branch = main
|
| 5 |
|
|
|
|
| 1 |
[submodule "sdxl-newdream-20"]
|
| 2 |
+
path = models/newdream-sdxl-20/
|
| 3 |
url = https://huggingface.co/stablediffusionapi/newdream-sdxl-20
|
| 4 |
branch = main
|
| 5 |
|
models/models--stablediffusionapi--newdream-sdxl-20/refs/main
DELETED
|
@@ -1 +0,0 @@
|
|
| 1 |
-
4bdd502bca7abd1ea57ee12fba0b0f23052958cc
|
|
|
|
|
|
models/models--stablediffusionapi--newdream-sdxl-20/snapshots/4bdd502bca7abd1ea57ee12fba0b0f23052958cc
DELETED
|
@@ -1 +0,0 @@
|
|
| 1 |
-
Subproject commit 4bdd502bca7abd1ea57ee12fba0b0f23052958cc
|
|
|
|
|
|
src/pipeline.py
CHANGED
|
@@ -7,9 +7,7 @@ from torch import Generator
|
|
| 7 |
|
| 8 |
def load_pipeline() -> StableDiffusionXLPipeline:
|
| 9 |
pipeline = StableDiffusionXLPipeline.from_pretrained(
|
| 10 |
-
"
|
| 11 |
-
revision="4bdd502bca7abd1ea57ee12fba0b0f23052958cc",
|
| 12 |
-
cache_dir="./models",
|
| 13 |
torch_dtype=torch.float16,
|
| 14 |
local_files_only=True,
|
| 15 |
).to("cuda")
|
|
|
|
| 7 |
|
| 8 |
def load_pipeline() -> StableDiffusionXLPipeline:
|
| 9 |
pipeline = StableDiffusionXLPipeline.from_pretrained(
|
| 10 |
+
"./models/newdream-sdxl-20",
|
|
|
|
|
|
|
| 11 |
torch_dtype=torch.float16,
|
| 12 |
local_files_only=True,
|
| 13 |
).to("cuda")
|