- complete(model,
- 'The object was blue all over, but also green all over, it was a',
+ f"Using {model_name} ({int(sum(p.numel() for p in model.parameters())/(1e6))}M parameters)"
+)
+
+print(
+ complete(
+ model,
+ tokenizer,
+ "The object was blue all over, but also green all over, it was a",