Mochi1

Runtime error

Yingqing commited on Apr 10, 2023

Commit

2eaa3f6

1 Parent(s): b884d2c

Update videocrafter_test.py

Files changed (1) hide show

videocrafter_test.py CHANGED Viewed

@@ -33,7 +33,7 @@ class Text2Video():
         self.lora_path_list = ['','models/videolora/lora_001_Loving_Vincent_style.ckpt',
                                 'models/videolora/lora_002_frozenmovie_style.ckpt',
                                 'models/videolora/lora_003_MakotoShinkaiYourName_style.ckpt',
-                                'models/videolora/lora_004_coco_style.ckpt']
         self.lora_trigger_word_list = ['','Loving Vincent style', 'frozenmovie style', 'MakotoShinkaiYourName style', 'coco style']
         model, _, _ = load_model(config, ckpt_path, gpu_id=0, inject_lora=False)
         self.model = model

         self.lora_path_list = ['','models/videolora/lora_001_Loving_Vincent_style.ckpt',
                                 'models/videolora/lora_002_frozenmovie_style.ckpt',
                                 'models/videolora/lora_003_MakotoShinkaiYourName_style.ckpt',
+                                'models/videolora/lora_004_coco_style_v2.ckpt']
         self.lora_trigger_word_list = ['','Loving Vincent style', 'frozenmovie style', 'MakotoShinkaiYourName style', 'coco style']
         model, _, _ = load_model(config, ckpt_path, gpu_id=0, inject_lora=False)
         self.model = model