Update sample.py

sample.py

@@ -91,6 +91,7 @@ def main(config: omegaconf.DictConfig) -> None:
     tokenizer=tokenizer,
     config=config, logger=False)
   pretrained.eval()
+  pretrained = pretrained.to('cuda')
 
   bindevaluator = BindEvaluator.load_from_checkpoint(
     config.guidance.classifier_checkpoint_path,
@@ -101,6 +102,7 @@ def main(config: omegaconf.DictConfig) -> None:
     d_k=64,
     d_v=128,
     d_inner=64)
+  bindevaluator = bindevaluator.to('cuda')
 
   samples = []
   for _ in tqdm(