fix noise_scheduler error (#1)

- change noise_scheduler to local (feed6af65c2c468f57b5e09cf793b69b77142319)

app.py

@@ -1,3 +1,19 @@
+import subprocess
+import sys
+
+def force_fix_huggingface_hub():
+    subprocess.check_call([
+        sys.executable,
+        "-m",
+        "pip",
+        "install",
+        "--no-deps",
+        "--force-reinstall",
+        "huggingface-hub==0.30.2"
+    ])
+
+force_fix_huggingface_hub()
+
 import gradio as gr
 from pathlib import Path
 
@@ -56,12 +72,14 @@ ckpt_path = hf_hub_download(
 hubert_models, _, _ = fairseq.checkpoint_utils.load_model_ensemble_and_task([ckpt_path])
 hubert_model = hubert_models[0].eval().to(device)
 
+scheduler_path = "./configs/models--stabilityai--stable-diffusion-2-1"
 scheduler = getattr(
     noise_schedulers,
     config["noise_scheduler"]["type"],
 ).from_pretrained(
-    config["noise_scheduler"]["name"],
+    scheduler_path,  
     subfolder="scheduler",
+    local_files_only=True
 )
 
 @torch.no_grad()

configs/models--stabilityai--stable-diffusion-2-1/scheduler/scheduler_config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "DDIMScheduler",
+  "_diffusers_version": "0.8.0",
+  "beta_end": 0.012,
+  "beta_schedule": "scaled_linear",
+  "beta_start": 0.00085,
+  "clip_sample": false,
+  "num_train_timesteps": 1000,
+  "prediction_type": "v_prediction",
+  "set_alpha_to_one": false,
+  "skip_prk_steps": true,
+  "steps_offset": 1,
+  "trained_betas": null
+}