dwb2023 commited on
Commit
33101b1
1 Parent(s): bf6c5c6

Update utils.py

Browse files

set HF_HUB_ENABLE_HF_TRANSFER = '1'

Files changed (1) hide show
  1. utils.py +2 -0
utils.py CHANGED
@@ -5,6 +5,8 @@ import spaces
5
  from huggingface_hub import login
6
  from transformers import BitsAndBytesConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration
7
 
 
 
8
  # Install required package
9
  def install_flash_attn():
10
  subprocess.run(
 
5
  from huggingface_hub import login
6
  from transformers import BitsAndBytesConfig, AutoModelForCausalLM, LlavaNextForConditionalGeneration, LlavaForConditionalGeneration, PaliGemmaForConditionalGeneration, Idefics2ForConditionalGeneration
7
 
8
+ os.environ = ['HF_HUB_ENABLE_HF_TRANSFER']
9
+
10
  # Install required package
11
  def install_flash_attn():
12
  subprocess.run(