Update prediction.py
Browse files- prediction.py +2 -0
prediction.py
CHANGED
|
@@ -46,6 +46,8 @@ def run_image_prediction(
|
|
| 46 |
config["model"]["params"]["condition_model_path"] = None
|
| 47 |
config["model"]["params"]["vqgan_model_path"] = None
|
| 48 |
|
|
|
|
|
|
|
| 49 |
# Instantiate model from config and move to device
|
| 50 |
model = instantiate_from_config(config.model).to(device)
|
| 51 |
|
|
|
|
| 46 |
config["model"]["params"]["condition_model_path"] = None
|
| 47 |
config["model"]["params"]["vqgan_model_path"] = None
|
| 48 |
|
| 49 |
+
os.chdir(os.path.dirname(model_ckpt_path))
|
| 50 |
+
|
| 51 |
# Instantiate model from config and move to device
|
| 52 |
model = instantiate_from_config(config.model).to(device)
|
| 53 |
|