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