Ransaka commited on
Commit
bb48876
1 Parent(s): 7e7b643

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -45,4 +45,7 @@ pipe = pipeline('text2text-generation', model=model, tokenizer=tokenizer)
45
  pipe(text)
46
  #[{'generated_text': 'theore, major tech are clamouring to replicate the generative AI chatbot developed by OpenAI using its AI'}]
47
 
 
 
 
48
  ```
 
45
  pipe(text)
46
  #[{'generated_text': 'theore, major tech are clamouring to replicate the generative AI chatbot developed by OpenAI using its AI'}]
47
 
48
+ print("Model memory usage: {:.2f} MB".format(pipe.model.get_memory_footprint()/1e6))
49
+ # 'Model memory usage: 1893.99 MB'
50
+
51
  ```