Skip to content

Commit

Permalink
refactor: minor linting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
jaydrennan committed Jan 25, 2024
1 parent cd23434 commit 8794fa6
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions imaginairy/enhancers/upscale.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,9 @@ def upscale_image(
import torchvision.transforms.functional as F
from spandrel import ImageModelDescriptor, ModelLoader


from imaginairy.utils.downloads import get_cached_url_path
from imaginairy.utils.tile_up import tile_process

device = device or get_device()
# device="cpu"

Expand All @@ -47,8 +47,7 @@ def upscale_image(
model_path = get_cached_url_path(model_url)
else:
model_path = upscaler_model
print("jay drennan")
print(model_path)

model = ModelLoader().load_from_file(model_path)
logger.info(
f"Upscaling from {img.width}x{img.height} to {img.width * model.scale}x{img.height * model.scale}"
Expand Down

0 comments on commit 8794fa6

Please sign in to comment.