Spaces:

SunderAli17
/

ToonMage

Running on Zero

SunderAli17 commited on Sep 13

Commit

7bfc133

•

1 Parent(s): d976653

Update ToonMage/fluxpipeline.py

Files changed (1) hide show

ToonMage/fluxpipeline.py CHANGED Viewed

@@ -84,8 +84,8 @@ class ToonMagePipeline(nn.Module):
         self.debug_img_list = []
     def load_pretrain(self, pretrain_path=None):
-        hf_hub_download('SunderAli17/SAK', 'toonmage_flux_v0.9.0.safetensors', local_dir='models')
-        ckpt_path = 'models/toonmage_flux_v0.9.0.safetensors'
         if pretrain_path is not None:
             ckpt_path = pretrain_path
         state_dict = load_file(ckpt_path)

         self.debug_img_list = []
     def load_pretrain(self, pretrain_path=None):
+        hf_hub_download('SunderAli17/SAK', 'toonmage_flux_v2.safetensors', local_dir='models')
+        ckpt_path = 'models/toonmage_flux_v2.safetensors'
         if pretrain_path is not None:
             ckpt_path = pretrain_path
         state_dict = load_file(ckpt_path)