Tongbo commited on
Commit
b3a5001
·
verified ·
1 Parent(s): b973fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import os
2
- os.environ["CUDA_VISIBLE_DEVICES"] = "1"
3
  import torch
4
  from flashsloth.constants import (
5
  IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN,
 
1
  import os
 
2
  import torch
3
  from flashsloth.constants import (
4
  IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN,