1inkusFace commited on
Commit
049edd5
·
verified ·
1 Parent(s): a6ceeee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ MAX_SEED = np.iinfo(np.int32).max
43
  hf_token = os.getenv("HF_TOKEN")
44
  openai_api_key = os.getenv("OPENAI_API_KEY")
45
  client = OpenAI(api_key=openai_api_key)
 
 
46
 
47
  system_prompt_t2v_path = "assets/system_prompt_t2v.txt"
48
  system_prompt_i2v_path = "assets/system_prompt_i2v.txt"
 
43
  hf_token = os.getenv("HF_TOKEN")
44
  openai_api_key = os.getenv("OPENAI_API_KEY")
45
  client = OpenAI(api_key=openai_api_key)
46
+ os.environ["SAFETENSORS_FAST_GPU"] = "1"
47
+ os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
48
 
49
  system_prompt_t2v_path = "assets/system_prompt_t2v.txt"
50
  system_prompt_i2v_path = "assets/system_prompt_i2v.txt"