BestWishYsh commited on
Commit
faa42fa
·
verified ·
1 Parent(s): 031d4d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -73,15 +73,15 @@ examples = [
73
  print(f"### Cleaning cached examples ...")
74
  os.system(f"rm -rf gradio_cached_examples/")
75
 
76
- device = torch.device('cuda:0')
77
 
78
  class MagicTimeController:
79
  def __init__(self):
80
  # config dirs
81
  self.basedir = os.getcwd()
82
- self.stable_diffusion_dir = os.path.join(self.basedir, "ckpts", "Base_Model")
83
- self.motion_module_dir = os.path.join(self.basedir, "ckpts", "Base_Model", "motion_module")
84
- self.personalized_model_dir = os.path.join(self.basedir, "ckpts", "DreamBooth")
85
  self.savedir = os.path.join(self.basedir, "outputs")
86
  os.makedirs(self.savedir, exist_ok=True)
87
 
 
73
  print(f"### Cleaning cached examples ...")
74
  os.system(f"rm -rf gradio_cached_examples/")
75
 
76
+ device = "cuda"
77
 
78
  class MagicTimeController:
79
  def __init__(self):
80
  # config dirs
81
  self.basedir = os.getcwd()
82
+ self.stable_diffusion_dir = os.path.join(self.basedir, model_path, "Base_Model")
83
+ self.motion_module_dir = os.path.join(self.basedir, model_path, "Base_Model", "motion_module")
84
+ self.personalized_model_dir = os.path.join(self.basedir, model_path, "DreamBooth")
85
  self.savedir = os.path.join(self.basedir, "outputs")
86
  os.makedirs(self.savedir, exist_ok=True)
87