debug code

This commit is contained in:
Seth Forsgren 2022-12-13 16:21:39 -08:00
parent 55651db0fc
commit 2fea921515
1 changed files with 0 additions and 5 deletions

View File

@ -36,8 +36,6 @@ class Model:
"riffusion/riffusion-model-v1", allow_patterns="*.png"
)
print(self._seed_images_dir)
def load(self):
# Load Riffusion model here and assign to self._model.
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@ -129,9 +127,6 @@ class Model:
# Load the seed image by ID
init_image_path = Path(self._seed_images_dir, f"seed_images/{inputs.seed_image_id}.png")
# Convert init_image_path blob to a file
print(init_image_path)
if not init_image_path.is_file():
return f"Invalid seed image: {inputs.seed_image_id}", 400
init_image = PIL.Image.open(str(init_image_path)).convert("RGB")