Hugging Face's logo

Spaces:
aidealab
/
AIdeaLab-VideoJP
Running on Zero

App Files Community
1
alfredplpl commited on
Commit
97c0741
·
verified ·
1 Parent(s): f4725aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ text_encoder = AutoModelForCausalLM.from_pretrained(
32
  text_encoder=text_encoder.to(device)
33
 
34
  transformer = CogVideoXTransformer3DModel.from_pretrained(
35
- os.environ["REPO_NAME"],
36
  torch_dtype=torch_dtype,
37
  token=os.environ['TOKEN']
38
  )
 
32
  text_encoder=text_encoder.to(device)
33
 
34
  transformer = CogVideoXTransformer3DModel.from_pretrained(
35
+ "aidealab/AIdeaLab-VideoJP",
36
  torch_dtype=torch_dtype,
37
  token=os.environ['TOKEN']
38
  )