patrickvonplaten commited on
Commit
19551f7
1 Parent(s): 7f458a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ if torch.cuda.is_available():
66
 
67
  # LCM Pipeline:
68
  pipe = LatentConsistencyModelPipeline(vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, scheduler=scheduler, safety_checker=safety_checker, feature_extractor=feature_extractor)
69
- pipe = pipe.to(device="cuda", dtype=DTYPE)
70
 
71
  if USE_TORCH_COMPILE:
72
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
66
 
67
  # LCM Pipeline:
68
  pipe = LatentConsistencyModelPipeline(vae=vae, text_encoder=text_encoder, tokenizer=tokenizer, unet=unet, scheduler=scheduler, safety_checker=safety_checker, feature_extractor=feature_extractor)
69
+ pipe = pipe.to(torch_device="cuda", torch_dtype=DTYPE)
70
 
71
  if USE_TORCH_COMPILE:
72
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)