Update README.md
Browse files
README.md
CHANGED
@@ -192,7 +192,7 @@ from safetensors.torch import safe_open # for loading .safetensors weights
|
|
192 |
# ---------------------------
|
193 |
# Model & Input Settings
|
194 |
# ---------------------------
|
195 |
-
model_path = "cycle_gan/
|
196 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
197 |
|
198 |
# Load your Generator (3→13 channels)
|
|
|
192 |
# ---------------------------
|
193 |
# Model & Input Settings
|
194 |
# ---------------------------
|
195 |
+
model_path = "cycle_gan/G_model.safetensors" # update to your model path
|
196 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
197 |
|
198 |
# Load your Generator (3→13 channels)
|