Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -49,7 +49,7 @@ model = Qwen2_5_VL_Draw(
|
|
| 49 |
|
| 50 |
# Load trained weights
|
| 51 |
from huggingface_hub import hf_hub_download
|
| 52 |
-
weights_path = hf_hub_download(repo_id="TESS-Computer/qwen-dit-draw", filename="
|
| 53 |
model.trajectory_head.load_state_dict(torch.load(weights_path, weights_only=True))
|
| 54 |
model = model.to("cuda").eval()
|
| 55 |
|
|
|
|
| 49 |
|
| 50 |
# Load trained weights
|
| 51 |
from huggingface_hub import hf_hub_download
|
| 52 |
+
weights_path = hf_hub_download(repo_id="TESS-Computer/qwen-dit-draw", filename="trajectory_head.pt")
|
| 53 |
model.trajectory_head.load_state_dict(torch.load(weights_path, weights_only=True))
|
| 54 |
model = model.to("cuda").eval()
|
| 55 |
|