saipuppala
/

diffusion_transformer

Model card Files Files and versions

saipuppala commited on 17 days ago

Commit

93fbf75

·

verified ·

1 Parent(s): 8a80e62

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Load fine-tuned model from Hugging Face
 model = PDETransformer.from_pretrained(
-    "saipuppala/pde-transformer-mc-s-diffusion-heat"
 ).to(device)
 model.eval()

 # Load fine-tuned model from Hugging Face
 model = PDETransformer.from_pretrained(
+    "saipuppala/diffusion_transformer"
 ).to(device)
 model.eval()