Spaces:

KumaPower
/

AvatarArtist

Running on Zero

刘虹雨 commited on Apr 1

Commit

10ae7b2

1 Parent(s): 23f9e19

update code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -477,8 +477,8 @@ def avatar_generation(items, save_path_base, video_path_input, source_type, is_s
     image_encoder.to(device)
     vae_triplane.to(device)
     dinov2.to(device)
-    ws_avg = ws_avg.to(device)
-    DiT_model = DiT_model.to(device)
     # Set up face verse for amimation
     base_coff = np.load(
         'pretrained_model/temp.npy').astype(

     image_encoder.to(device)
     vae_triplane.to(device)
     dinov2.to(device)
+    ws_avg.to(device)
+    DiT_model.to(device)
     # Set up face verse for amimation
     base_coff = np.load(
         'pretrained_model/temp.npy').astype(