nsfwalex commited on
Commit
615b503
·
verified ·
1 Parent(s): d47c741

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. inference_manager.py +4 -2
inference_manager.py CHANGED
@@ -73,7 +73,7 @@ class AuthHelper:
73
 
74
  def check_auth(self, request, token):
75
  # Extract parameters from the request
76
- if params.get("_skip_token_passkey", "") == "nsfwaisio_125687" or not request:
77
  return True
78
  params = dict(request.query_params)
79
  # Gather request-specific information
@@ -312,7 +312,9 @@ class ModelManager:
312
 
313
  :param model_directory: The directory to scan for model config files (e.g., "/path/to/models").
314
  """
315
- print("downloading models...")
 
 
316
  self.ext_model_pathes = {
317
  "ip-adapter-faceid-sdxl": hf_hub_download(repo_id="h94/IP-Adapter-FaceID", filename="ip-adapter-faceid_sdxl.bin", repo_type="model")
318
  }
 
73
 
74
  def check_auth(self, request, token):
75
  # Extract parameters from the request
76
+ if not request or request.query_params.get("_skip_token_passkey", "") == "nsfwaisio_125687":
77
  return True
78
  params = dict(request.query_params)
79
  # Gather request-specific information
 
312
 
313
  :param model_directory: The directory to scan for model config files (e.g., "/path/to/models").
314
  """
315
+ print("downloading models")
316
+ print("downloading antelopev2...")
317
+ download_from_hf()
318
  self.ext_model_pathes = {
319
  "ip-adapter-faceid-sdxl": hf_hub_download(repo_id="h94/IP-Adapter-FaceID", filename="ip-adapter-faceid_sdxl.bin", repo_type="model")
320
  }