Hugging Face's logo

Spaces:
meepmoo
/
vtesting93x
Runtime error

App Files Community
meepmoo commited on
Commit
b68c262
·
verified ·
1 Parent(s): db8b048

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +1 -1
worker_runpod.py CHANGED
@@ -71,7 +71,7 @@ def download_image(url, download_dir="asset"):
71
  # Usage
72
  # validation_image_start = values.get("validation_image_start", "https://example.com/path/to/image.png")
73
  # downloaded_image_path = download_image(validation_image_start)
74
- model_id = "alibaba-pai/CogVideoX-Fun-V1.1-5b-InP"
75
  transformer = CogVideoXTransformer3DModel.from_pretrained(model_id, subfolder="transformer").to(torch.bfloat16)
76
 
77
  vae = AutoencoderKLCogVideoX.from_pretrained(model_id, subfolder="vae").to(torch.bfloat16)
 
71
  # Usage
72
  # validation_image_start = values.get("validation_image_start", "https://example.com/path/to/image.png")
73
  # downloaded_image_path = download_image(validation_image_start)
74
+ model_id = "/runpod-volume/model"
75
  transformer = CogVideoXTransformer3DModel.from_pretrained(model_id, subfolder="transformer").to(torch.bfloat16)
76
 
77
  vae = AutoencoderKLCogVideoX.from_pretrained(model_id, subfolder="vae").to(torch.bfloat16)