ShaoTengLiu commited on
Commit
9eb490d
1 Parent(s): 2d2f0f8
Files changed (2) hide show
  1. app.py +1 -1
  2. trainer.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from trainer import Trainer
16
 
17
  TITLE = '# [Tune-A-Video](https://tuneavideo.github.io/) UI'
18
 
19
- ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
20
  SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
21
  GPU_DATA = getoutput('nvidia-smi')
22
  SHARED_UI_WARNING = f'''## Attention - Training doesn't work in this shared UI. You can duplicate and use it with a paid private T4 GPU.
 
16
 
17
  TITLE = '# [Tune-A-Video](https://tuneavideo.github.io/) UI'
18
 
19
+ ORIGINAL_SPACE_ID = 'video-p2p-library/Video-P2P-Demo'
20
  SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
21
  GPU_DATA = getoutput('nvidia-smi')
22
  SHARED_UI_WARNING = f'''## Attention - Training doesn't work in this shared UI. You can duplicate and use it with a paid private T4 GPU.
trainer.py CHANGED
@@ -20,7 +20,7 @@ from utils import save_model_card
20
  sys.path.append('Tune-A-Video')
21
 
22
  URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/Tune-A-Video-library/share/YjTcaNJmKyeHFpMBioHhzBcTzCYddVErEk'
23
- ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
24
  SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
25
 
26
 
 
20
  sys.path.append('Tune-A-Video')
21
 
22
  URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/Tune-A-Video-library/share/YjTcaNJmKyeHFpMBioHhzBcTzCYddVErEk'
23
+ ORIGINAL_SPACE_ID = 'video-p2p-library/Video-P2P-Demo'
24
  SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
25
 
26