davanstrien HF Staff commited on
Commit
a44a287
·
1 Parent(s): 82483a0

Enable HF_TRANSFER for model loading

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  from transformers import AutoProcessor, AutoModelForImageTextToText, pipeline
7
  import spaces
8
 
 
9
  # --- Global Model and Processor ---
10
  MODELS = {}
11
  PROCESSORS = {}
 
6
  from transformers import AutoProcessor, AutoModelForImageTextToText, pipeline
7
  import spaces
8
 
9
+ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1" # turn on HF_TRANSFER
10
  # --- Global Model and Processor ---
11
  MODELS = {}
12
  PROCESSORS = {}