Maybe fix JIT...

2c580c72 · novelailab · 1861450d · 2c580c72
Commit 2c580c72 authored Sep 29, 2022 by novelailab
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

hydra_node/models.py hydra_node/models.py +1 -1

No files found.
--- a/hydra_node/models.py
+++ b/hydra_node/models.py
@@ -311,7 +311,7 @@ class StableDiffusionModel(nn.Module):
        config = self.get_default_config
        config.steps = 1
        config.prompt = ""
-        self.sample(config, ema=ema)
+        self.sample(config)

    def from_folder(self, folder):
        folder = Path(folder)