rafaaa2105 commited on
Commit
c2b6d1e
1 Parent(s): a3de98d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,9 +8,12 @@ import spaces
8
  model_list = [model.strip() for model in os.environ.get("MODELS").split(",")]
9
  lora_list = [model.strip() for model in os.environ.get("LORAS").split(",")]
10
 
 
 
11
  models = {}
12
  for model_name in model_list:
13
  try:
 
14
  models[model_name] = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16).to("cuda")
15
  except Exception as e:
16
  print(f"Error loading model {model_name}: {e}")
 
8
  model_list = [model.strip() for model in os.environ.get("MODELS").split(",")]
9
  lora_list = [model.strip() for model in os.environ.get("LORAS").split(",")]
10
 
11
+ print("Detected " + len(model_list) + " on models and " + len(lora_list) + "LoRAs.")
12
+
13
  models = {}
14
  for model_name in model_list:
15
  try:
16
+ print(f"Loading {model_name}...")
17
  models[model_name] = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16).to("cuda")
18
  except Exception as e:
19
  print(f"Error loading model {model_name}: {e}")