Yaron Koresh commited on
Commit
0c8256b
·
verified ·
1 Parent(s): 9f25293

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,7 +48,7 @@ else:
48
  dtype = torch.float16
49
 
50
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
51
- adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
52
 
53
  # variable data
54
 
@@ -109,7 +109,7 @@ function custom(){
109
  # torch pipes
110
 
111
  image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
112
- pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter).to(device)
113
  pipe.scheduler = EulerAncestralDiscreteScheduler(
114
  base,
115
  subfolder="scheduler",
 
48
  dtype = torch.float16
49
 
50
  base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
51
+ adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device, safety_checker=None)
52
 
53
  # variable data
54
 
 
109
  # torch pipes
110
 
111
  image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
112
+ pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter, safety_checker=None).to(device)
113
  pipe.scheduler = EulerAncestralDiscreteScheduler(
114
  base,
115
  subfolder="scheduler",