Spaces:

Ruurd
/

lad

Running on Zero

Ruurd commited on Apr 8

Commit

617b180

1 Parent(s): 0e3f268

Try again

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ token_probabilities = np.array([token_probs_dict[str(i)] for i in range(len(toke
 def load_model():
     ckpt_path = hf_hub_download(
-        repo_id="ruurd/diffusion-llama",
         filename="diffusion-model.pth",
         token=os.getenv("HF_TOKEN")
     )

 def load_model():
     ckpt_path = hf_hub_download(
+        repo_id="ruurd/tini_model",
         filename="diffusion-model.pth",
         token=os.getenv("HF_TOKEN")
     )