Yaron Koresh commited on
Commit
0c5ef34
·
verified ·
1 Parent(s): 762a623

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -47,7 +47,7 @@ root.addHandler(handler2)
47
  # constant data
48
 
49
  dtype = torch.float16
50
- device = "cuda"
51
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
52
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
53
 
@@ -85,7 +85,6 @@ footer {
85
  }
86
  #col-container {
87
  margin: 0 auto;
88
- max-width: 15cm;
89
  }
90
  .image-container {
91
  aspect-ratio: """,str(width),"/",str(height),""" !important;
 
47
  # constant data
48
 
49
  dtype = torch.float16
50
+ device = torch.cuda.is_available() ? "cuda" : "cpu"
51
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
52
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
53
 
 
85
  }
86
  #col-container {
87
  margin: 0 auto;
 
88
  }
89
  .image-container {
90
  aspect-ratio: """,str(width),"/",str(height),""" !important;