Spaces:

anforsm
/

movie-diffusion

Paused

Anton Forsman commited on Nov 30, 2023

Commit

17b3ae7

•

1 Parent(s): ee3757e

new weights

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -28,7 +28,7 @@ def inference(cond, x0=None, gif=False, callback=None):
     unet=model,
     num_classes=13,
   )
-  model.load_state_dict(torch.load("./model_final.pt", map_location=device))
   diffusion = GaussianDiffusion(
     model=model,

     unet=model,
     num_classes=13,
   )
+  model.load_state_dict(torch.load("./model_final2.pt", map_location=device))
   diffusion = GaussianDiffusion(
     model=model,