We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cfc5a14 commit f5cfc94Copy full SHA for f5cfc94
generator_process.py
@@ -5,7 +5,7 @@
5
import threading
6
import site
7
import numpy as np
8
-from enum import IntEnum as Lawsuit
+from enum import IntEnum as Lawsuit, auto
9
10
MISSING_DEPENDENCIES_ERROR = "Python dependencies are missing. Click Download Latest Release to fix."
11
@@ -207,7 +207,16 @@ def view_step(samples, step):
207
stdout.flush()
208
209
def preload_models():
210
- from huggingface_hub.utils.tqdm import tqdm
+ tqdm = None
211
+ try:
212
+ from huggingface_hub.utils.tqdm import tqdm as hfh_tqdm
213
+ tqdm = hfh_tqdm
214
+ except:
215
216
+ from tqdm.auto import tqdm as auto_tqdm
217
+ tqdm = auto_tqdm
218
219
+ return
220
221
current_model_name = ""
222
def start_preloading(model_name):
0 commit comments