More logging

app.py

@@ -45,6 +45,12 @@ vardecoder_model = AutoModelForCausalLM.from_pretrained(
 )
 print("Loaded vardecoder model successfully.")
 
+print(f"Model device: {next(vardecoder_model.parameters()).device}")
+print(f"Model dtype: {next(vardecoder_model.parameters()).dtype}")
+print(f"Model is meta: {next(vardecoder_model.parameters()).is_meta}")
+print(f"Model parameters: {sum(p.numel() for p in vardecoder_model.parameters() if p.requires_grad):,}")
+
+
 # Check memory after first model
 print(f"GPU memory after vardecoder:")
 print(f"Allocated: {torch.cuda.memory_allocated() / 1024**3:.2f} GB")