add additional debug info

app.py

@@ -87,7 +87,8 @@ def infer(
     print("Loading Models\n")
     for idx in model_idx:
         model_name = MODEL_NAMES[idx]
-        if models[model_name] == None: 
+        if models[model_name] == None:
+            print ("Initializing " + model_name)
             tokenizer = BloomTokenizerFast.from_pretrained(model_name)
             model = DistributedBloomForCausalLM.from_pretrained(model_name, torch_dtype=TORCH_DTYPE)
             model = model.to(DEVICE)