liuyuan-pal commited on
Commit
864884a
1 Parent(s): a500583
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -11,6 +11,12 @@ from omegaconf import OmegaConf
11
  from ldm.util import add_margin, instantiate_from_config
12
  from sam_utils import sam_init, sam_out_nosave
13
 
 
 
 
 
 
 
14
  _TITLE = '''SyncDreamer: Generating Multiview-consistent Images from a Single-view Image'''
15
  _DESCRIPTION = '''
16
  <div>
 
11
  from ldm.util import add_margin, instantiate_from_config
12
  from sam_utils import sam_init, sam_out_nosave
13
 
14
+ import torch
15
+ print(f"Is CUDA available: {torch.cuda.is_available()}")
16
+ # True
17
+ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
18
+ # Tesla T4
19
+
20
  _TITLE = '''SyncDreamer: Generating Multiview-consistent Images from a Single-view Image'''
21
  _DESCRIPTION = '''
22
  <div>