charliebaby2023 commited on
Commit
9c95830
·
verified ·
1 Parent(s): a83c936

Update handle_models.py

Browse files
Files changed (1) hide show
  1. handle_models.py +4 -0
handle_models.py CHANGED
@@ -4,9 +4,13 @@ from all_models import models
4
  import asyncio
5
  from externalmod import gr_Interface_load, save_image, randomize_seed
6
  import os
 
 
7
  HF_TOKEN = os.getenv("HF_TOKEN")
8
  default_models = models[:num_models]
9
 
 
 
10
  def load_fn(models,HF_TOKEN):
11
  global models_load
12
  models_load = {}
 
4
  import asyncio
5
  from externalmod import gr_Interface_load, save_image, randomize_seed
6
  import os
7
+ from threading import RLock
8
+ lock = RLock()
9
  HF_TOKEN = os.getenv("HF_TOKEN")
10
  default_models = models[:num_models]
11
 
12
+
13
+
14
  def load_fn(models,HF_TOKEN):
15
  global models_load
16
  models_load = {}