Scheduler_Update (#19)
Browse files- update scheduler (fced3be63f70057c6c8f1d760b5443d8a123c1d0)
Co-authored-by: Yan Shu <[email protected]>
- model_index.json +1 -1
- scheduler/scheduler_config.json +25 -3
    	
        model_index.json
    CHANGED
    
    | @@ -3,7 +3,7 @@ | |
| 3 | 
             
              "_diffusers_version": "0.32.1",
         | 
| 4 | 
             
              "scheduler": [
         | 
| 5 | 
             
                "diffusers",
         | 
| 6 | 
            -
                " | 
| 7 | 
             
              ],
         | 
| 8 | 
             
              "text_encoder": [
         | 
| 9 | 
             
                "transformers",
         | 
|  | |
| 3 | 
             
              "_diffusers_version": "0.32.1",
         | 
| 4 | 
             
              "scheduler": [
         | 
| 5 | 
             
                "diffusers",
         | 
| 6 | 
            +
                "UniPCMultistepScheduler"
         | 
| 7 | 
             
              ],
         | 
| 8 | 
             
              "text_encoder": [
         | 
| 9 | 
             
                "transformers",
         | 
    	
        scheduler/scheduler_config.json
    CHANGED
    
    | @@ -1,6 +1,28 @@ | |
| 1 | 
             
            {
         | 
| 2 | 
            -
              "_class_name": " | 
| 3 | 
            -
              "_diffusers_version": "0. | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 4 | 
             
              "num_train_timesteps": 1000,
         | 
| 5 | 
            -
              " | 
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
|  | |
| 6 | 
             
            }
         | 
|  | |
| 1 | 
             
            {
         | 
| 2 | 
            +
              "_class_name": "UniPCMultistepScheduler",
         | 
| 3 | 
            +
              "_diffusers_version": "0.34.0.dev0",
         | 
| 4 | 
            +
              "beta_end": 0.02,
         | 
| 5 | 
            +
              "beta_schedule": "linear",
         | 
| 6 | 
            +
              "beta_start": 0.0001,
         | 
| 7 | 
            +
              "disable_corrector": [],
         | 
| 8 | 
            +
              "dynamic_thresholding_ratio": 0.995,
         | 
| 9 | 
            +
              "final_sigmas_type": "zero",
         | 
| 10 | 
            +
              "flow_shift": 3.0,
         | 
| 11 | 
            +
              "lower_order_final": true,
         | 
| 12 | 
             
              "num_train_timesteps": 1000,
         | 
| 13 | 
            +
              "predict_x0": true,
         | 
| 14 | 
            +
              "prediction_type": "flow_prediction",
         | 
| 15 | 
            +
              "rescale_betas_zero_snr": false,
         | 
| 16 | 
            +
              "sample_max_value": 1.0,
         | 
| 17 | 
            +
              "solver_order": 2,
         | 
| 18 | 
            +
              "solver_p": null,
         | 
| 19 | 
            +
              "solver_type": "bh2",
         | 
| 20 | 
            +
              "steps_offset": 0,
         | 
| 21 | 
            +
              "thresholding": false,
         | 
| 22 | 
            +
              "timestep_spacing": "linspace",
         | 
| 23 | 
            +
              "trained_betas": null,
         | 
| 24 | 
            +
              "use_beta_sigmas": false,
         | 
| 25 | 
            +
              "use_exponential_sigmas": false,
         | 
| 26 | 
            +
              "use_flow_sigmas": true,
         | 
| 27 | 
            +
              "use_karras_sigmas": false
         | 
| 28 | 
             
            }
         | 
