Spaces:
Runtime error
Runtime error
Raman Dutt
commited on
Commit
·
d86745c
1
Parent(s):
cab6ad5
minor fix
Browse files
app.py
CHANGED
|
@@ -217,7 +217,6 @@ def predict(
|
|
| 217 |
num_inference_steps=75,
|
| 218 |
device="0",
|
| 219 |
OUTPUT_DIR="OUTPUT",
|
| 220 |
-
PIPELINE_DICT=PIPELINE_DICT,
|
| 221 |
):
|
| 222 |
|
| 223 |
NUM_TUNABLE_PARAMS = {
|
|
@@ -233,8 +232,6 @@ def predict(
|
|
| 233 |
|
| 234 |
cuda_device = f"cuda:{device}" if torch.cuda.is_available() else "cpu"
|
| 235 |
|
| 236 |
-
|
| 237 |
-
#sd_pipeline = PIPELINE_DICT[unet_pretraining_type]
|
| 238 |
print("Loading Pipeline for {} Fine-Tuning".format(unet_pretraining_type))
|
| 239 |
sd_pipeline_norm = loadSDModel(
|
| 240 |
unet_pretraining_type=unet_pretraining_type,
|
|
|
|
| 217 |
num_inference_steps=75,
|
| 218 |
device="0",
|
| 219 |
OUTPUT_DIR="OUTPUT",
|
|
|
|
| 220 |
):
|
| 221 |
|
| 222 |
NUM_TUNABLE_PARAMS = {
|
|
|
|
| 232 |
|
| 233 |
cuda_device = f"cuda:{device}" if torch.cuda.is_available() else "cpu"
|
| 234 |
|
|
|
|
|
|
|
| 235 |
print("Loading Pipeline for {} Fine-Tuning".format(unet_pretraining_type))
|
| 236 |
sd_pipeline_norm = loadSDModel(
|
| 237 |
unet_pretraining_type=unet_pretraining_type,
|