Hugging Face's logo

Spaces:
ezeanubis
/
voyager
Runtime error

App Files Community
ezeanubis commited on
Commit
c6f0b4e
·
verified ·
1 Parent(s): 2e995e3

Update tools/infer.py

Browse files
Files changed (1) hide show
  1. tools/infer.py +2 -1
tools/infer.py CHANGED
@@ -6,7 +6,8 @@ from PIL import Image
6
  from huggingface_hub import hf_hub_download
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
- MODEL_REPO = "tencent/HunyuanWorld-Voyager"
 
10
 
11
  def load_model():
12
  print(f"Descargando y cargando modelo desde {MODEL_REPO} ...")
 
6
  from huggingface_hub import hf_hub_download
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
9
+ MODEL_REPO = "stabilityai/stable-diffusion-2"
10
+
11
 
12
  def load_model():
13
  print(f"Descargando y cargando modelo desde {MODEL_REPO} ...")