Upload folder using huggingface_hub

.idea/workspace.xml

@@ -54,7 +54,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1760056680813</updated>
-      <workItem from="1760056681869" duration="2999000" />
+      <workItem from="1760056681869" duration="3537000" />
     </task>
     <servers />
   </component>

app.py

@@ -50,12 +50,19 @@ def center_crop(image):
 ##### load model
 config = "configs/pipelines/stage_2_base.py"
 config = Config.fromfile(config)
-model = BUILDER.build(config.model).cuda().bfloat16().eval()
+model = BUILDER.build(config.model).eval()
 checkpoint_path = "checkpoints/Puffin-Base.pth"
 checkpoint = torch.load(checkpoint_path, map_location='cpu')
 _ = model.load_state_dict(checkpoint, strict=False)
 _ = model.vae.load_state_dict(torch.load('checkpoints/vae.pth', map_location='cpu'), strict=True)
 
+
+if torch.cuda.is_available():
+    harmon_model = model.to(torch.bfloat16).cuda()
+else:
+    harmon_model = model.to(torch.float32)
+
+
 def fig_to_image(fig):
     buf = io.BytesIO()
     fig.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)