weathon commited on
Commit
e1f3a33
·
1 Parent(s): 18f8353
app.py CHANGED
@@ -102,6 +102,7 @@ def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step,
102
  nag_alpha=nag_alpha,
103
  nag_tau=nag_tau,
104
  guidance_scale=0.0,
 
105
  ).images[0]
106
  nag_path = f"images/{uuid.uuid4().hex}_nag.png"
107
  output_nag.save(nag_path)
@@ -112,6 +113,7 @@ def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step,
112
  num_inference_steps=nag_step,
113
  nag_scale=0.0,
114
  guidance_scale=0.0,
 
115
  ).images[0]
116
  normal_path = f"images/{uuid.uuid4().hex}_normal.png"
117
  output_normal.save(normal_path)
 
102
  nag_alpha=nag_alpha,
103
  nag_tau=nag_tau,
104
  guidance_scale=0.0,
105
+ generator=torch.Generator(device="cuda").manual_seed(seed),
106
  ).images[0]
107
  nag_path = f"images/{uuid.uuid4().hex}_nag.png"
108
  output_nag.save(nag_path)
 
113
  num_inference_steps=nag_step,
114
  nag_scale=0.0,
115
  guidance_scale=0.0,
116
+ generator=torch.Generator(device="cuda").manual_seed(seed),
117
  ).images[0]
118
  normal_path = f"images/{uuid.uuid4().hex}_normal.png"
119
  output_normal.save(normal_path)
src/__pycache__/__init__.cpython-310.pyc CHANGED
Binary files a/src/__pycache__/__init__.cpython-310.pyc and b/src/__pycache__/__init__.cpython-310.pyc differ
 
src/__pycache__/processor.cpython-310.pyc CHANGED
Binary files a/src/__pycache__/processor.cpython-310.pyc and b/src/__pycache__/processor.cpython-310.pyc differ
 
src/__pycache__/sd3_pipeline.cpython-310.pyc CHANGED
Binary files a/src/__pycache__/sd3_pipeline.cpython-310.pyc and b/src/__pycache__/sd3_pipeline.cpython-310.pyc differ
 
vsfwan/__pycache__/__init__.cpython-310.pyc CHANGED
Binary files a/vsfwan/__pycache__/__init__.cpython-310.pyc and b/vsfwan/__pycache__/__init__.cpython-310.pyc differ
 
vsfwan/__pycache__/pipeline.cpython-310.pyc CHANGED
Binary files a/vsfwan/__pycache__/pipeline.cpython-310.pyc and b/vsfwan/__pycache__/pipeline.cpython-310.pyc differ
 
vsfwan/__pycache__/processor.cpython-310.pyc CHANGED
Binary files a/vsfwan/__pycache__/processor.cpython-310.pyc and b/vsfwan/__pycache__/processor.cpython-310.pyc differ