Download scheduler config from repo

2023-11-09 00:07:31 +01:00
parent 4c438e1ee6
commit 73aa2d11d3
2 changed files with 2 additions and 20 deletions
@@ -36,7 +36,8 @@ class CoreMLSamplerLCM(CoreMLSampler):

    def __init__(self):
        self.scheduler = LCMScheduler.from_pretrained(
-            os.path.join(os.path.dirname(__file__), "scheduler_config.json")
+            "SimianLuo/LCM_Dreamshaper_v7",
+            subfolder="scheduler"
        )

    def sample(
@@ -1,19 +0,0 @@
-{
-  "_class_name": "LCMScheduler",
-  "_diffusers_version": "0.22.0.dev0",
-  "beta_end": 0.012,
-  "beta_schedule": "scaled_linear",
-  "beta_start": 0.00085,
-  "clip_sample": true,
-  "clip_sample_range": 1.0,
-  "dynamic_thresholding_ratio": 0.995,
-  "num_train_timesteps": 1000,
-  "prediction_type": "epsilon",
-  "rescale_betas_zero_snr": false,
-  "sample_max_value": 1.0,
-  "set_alpha_to_one": true,
-  "steps_offset": 0,
-  "thresholding": false,
-  "timestep_spacing": "leading",
-  "trained_betas": null
-}