update swarovski

FateZero/video_diffusion/models/resnet.py

@@ -27,17 +27,20 @@ class PseudoConv3d(nn.Conv2d):
             
             
         if 'lora' in model_config.keys() :
-            self.conv_temporal = (
-                LoRALinearLayer(
-                    out_channels,
-                    out_channels,
-                    rank=model_config['lora'],
-                    stride=temporal_stride
-                    
+            if model_config['lora'] is not None:
+                self.conv_temporal = (
+                    LoRALinearLayer(
+                        out_channels,
+                        out_channels,
+                        rank=model_config['lora'],
+                        stride=temporal_stride
+                        
+                    )
+                    if kernel_size > 1
+                    else None
                 )
-                if kernel_size > 1
-                else None
-            )
+            else:
+                self.conv_temporal = None
         else:
             self.conv_temporal = (
                 nn.Conv1d(

example.py

@@ -1,3 +1,4 @@
+num_steps = 10
 style_example = [
             [
                 'CompVis/stable-diffusion-v1-4',
@@ -8,7 +9,7 @@ style_example = [
                 0.8,
                 "watercolor",
                 10,
-                10,
+                num_steps,
                 7.5,
                 # input video argument
                 None, 0, 8, 1, 0,0,0,0
@@ -23,7 +24,7 @@ style_example = [
                 0.5,
                 'van gogh',
                 10,
-                10,
+                num_steps,
                 7.5,
                 None, 0, 8, 1, 0,0,0,0
             ],
@@ -36,7 +37,7 @@ style_example = [
                 0.9,
                 'Ukiyo-e',
                 10,
-                10,
+                num_steps,
                 7.5,
                 None, 0, 8, 1, 0,0,0,0
             ],
@@ -49,7 +50,21 @@ style_example = [
                 0.9,
                 'Makoto Shinkai',
                 10,
+                num_steps,
+                7.5,
+                None, 0, 8, 1, 0,0,0,0
+            ],
+
+            [
+                'CompVis/stable-diffusion-v1-4',
+                'FateZero/data/attribute/swan_swarov',
+                'a black swan with a red beak swimming in a river near a wall and bushes',
+                'a Swarovski crystal swan with a red beak swimming in a river near a wall and bushes',
+                0.8,
+                0.6,
+                'Swarovski crystal',
                 10,
+                num_steps,
                 7.5,
                 None, 0, 8, 1, 0,0,0,0
             ],