dwb2023 commited on
Commit
79ab92b
1 Parent(s): c8190f8

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -5,7 +5,7 @@ import spaces
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'] = '1'
9
 
10
  # Install required package
11
  def install_flash_attn():
 
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"] = "1"
9
 
10
  # Install required package
11
  def install_flash_attn():