Hugging Face's logo

Spaces:
SunderAli17
/
ToonMage
Running on Zero

App Files Community
SunderAli17 commited on
Commit
7a00d58
·
verified ·
1 Parent(s): 299d327

Update toonmage/fluxpipeline.py

Browse files
Files changed (1) hide show
  1. toonmage/fluxpipeline.py +2 -2
toonmage/fluxpipeline.py CHANGED
@@ -84,8 +84,8 @@ class ToonMagePipeline(nn.Module):
84
  self.debug_img_list = []
85
 
86
  def load_pretrain(self, pretrain_path=None):
87
- hf_hub_download('SunderAli17/SAK', 'toonmage_flux_v2.safetensors')
88
- ckpt_path = 'toonmage_flux_v2.safetensors'
89
  if pretrain_path is not None:
90
  ckpt_path = pretrain_path
91
  state_dict = load_file(ckpt_path)
 
84
  self.debug_img_list = []
85
 
86
  def load_pretrain(self, pretrain_path=None):
87
+ hf_hub_download('SunderAli17/SAK', 'toonmage_flux_v2.safetensors', local_dir ='models')
88
+ ckpt_path = 'models/toonmage_flux_v2.safetensors'
89
  if pretrain_path is not None:
90
  ckpt_path = pretrain_path
91
  state_dict = load_file(ckpt_path)