Spaces:
Runtime error
Runtime error
Commit
·
71ed815
1
Parent(s):
3ff99e6
Update app.py
Browse files
app.py
CHANGED
@@ -17,17 +17,10 @@ SAMPLE_IMAGE = 'weights/Sample.png'
|
|
17 |
|
18 |
def load_pipeline_w_lora() :
|
19 |
|
20 |
-
|
21 |
-
unet = UNet2DConditionModel.from_pretrained(
|
22 |
-
MODEL_REPO,
|
23 |
-
subfolder="unet",
|
24 |
-
revision=None
|
25 |
-
)
|
26 |
-
|
27 |
# Load pipeline
|
28 |
pipeline = DiffusionPipeline.from_pretrained(
|
29 |
MODEL_REPO,
|
30 |
-
unet=unet,
|
31 |
revision=None,
|
32 |
torch_dtype=torch.float32,
|
33 |
)
|
|
|
17 |
|
18 |
def load_pipeline_w_lora() :
|
19 |
|
20 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
# Load pipeline
|
22 |
pipeline = DiffusionPipeline.from_pretrained(
|
23 |
MODEL_REPO,
|
|
|
24 |
revision=None,
|
25 |
torch_dtype=torch.float32,
|
26 |
)
|