Update NonReference_RGBN_x4/load.py
Browse files
NonReference_RGBN_x4/load.py
CHANGED
|
@@ -15,7 +15,10 @@ config = {
|
|
| 15 |
"denoiser_settings": {
|
| 16 |
"linear_start": 0.0015,
|
| 17 |
"linear_end": 0.0155,
|
| 18 |
-
"timesteps": 1000
|
|
|
|
|
|
|
|
|
|
| 19 |
},
|
| 20 |
"first_stage_config": {
|
| 21 |
"embed_dim": 4,
|
|
|
|
| 15 |
"denoiser_settings": {
|
| 16 |
"linear_start": 0.0015,
|
| 17 |
"linear_end": 0.0155,
|
| 18 |
+
"timesteps": 1000,
|
| 19 |
+
"sampling_eta": 1.0,
|
| 20 |
+
"sampling_steps":200,
|
| 21 |
+
"sampling_temperature":1.,
|
| 22 |
},
|
| 23 |
"first_stage_config": {
|
| 24 |
"embed_dim": 4,
|