Spaces:

DawnC
/

PawMatchAI

Running on Zero

DawnC commited on 27 days ago

Commit

d37a573

verified ·

1 Parent(s): 4bef216

Update style_transfer.py

Files changed (1) hide show

style_transfer.py CHANGED Viewed

@@ -128,7 +128,6 @@ class DogStyleTransfer:
                     cache_dir=self.model_cache_dir,
                     torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
                     safety_checker=None,  # Remove safety checker to improve speed
-                    revision="main"
                 )
                 if self.device == "cuda":

                     cache_dir=self.model_cache_dir,
                     torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
                     safety_checker=None,  # Remove safety checker to improve speed
                 )
                 if self.device == "cuda":