chansung commited on
Commit
3981b8f
1 Parent(s): 7f0ed3e

Update vid2persona/init.py

Browse files
Files changed (1) hide show
  1. vid2persona/init.py +1 -0
vid2persona/init.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import google.auth
 
3
 
4
  # https://huggingface.co/blog/inference-pro
5
  ALLOWED_LLM_FOR_HF_PRO_ACCOUNTS = [
 
1
  import os
2
  import google.auth
3
+ from transformers import AutoModelForCausalLM, AutoTokenizer
4
 
5
  # https://huggingface.co/blog/inference-pro
6
  ALLOWED_LLM_FOR_HF_PRO_ACCOUNTS = [