jwkirchenbauer commited on
Commit
0491786
·
1 Parent(s): ea229c1

disable hf transfer feature

Browse files
Files changed (1) hide show
  1. demo_watermark.py +2 -0
demo_watermark.py CHANGED
@@ -24,6 +24,8 @@ import gradio as gr
24
 
25
  import torch
26
 
 
 
27
  from transformers import (AutoTokenizer,
28
  AutoModelForSeq2SeqLM,
29
  AutoModelForCausalLM,
 
24
 
25
  import torch
26
 
27
+ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
28
+
29
  from transformers import (AutoTokenizer,
30
  AutoModelForSeq2SeqLM,
31
  AutoModelForCausalLM,