french version 1.00

app.py

@@ -31,11 +31,7 @@ def gpu_decorator(func):
     else:
         return func
 
-device = (
-    "cpu"
-    if torch.cuda.is_available()
-    else "mps" if torch.backends.mps.is_available() else "cpu"
-)
+device = device = "cuda" if torch.cuda.is_available() else "cpu"
 
 print(f"Using {device} device")
 

model/utils.py

@@ -572,9 +572,9 @@ def load_checkpoint(model, ckpt_path, device, use_ema = True):
         if ckpt_type == "safetensors":
             ema_model.load_state_dict(checkpoint)
         else:
-            ema_model.load_state_dict(checkpoint['ema_model_state_dict'])
+            ema_model.load_state_dict(checkpoint['ema_model_state_dict'], strict=False)
         ema_model.copy_params_from_ema_to_model()
     else:
-        model.load_state_dict(checkpoint['model_state_dict'])
+        model.load_state_dict(checkpoint['model_state_dict'], strict=False)
         
     return model