Menyu commited on
Commit
82120d1
·
verified ·
1 Parent(s): 2bf52cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -21,6 +21,7 @@ if torch.cuda.is_available():
21
  add_watermarker=False
22
  )
23
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
 
24
  pipe.to("cuda")
25
 
26
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
21
  add_watermarker=False
22
  )
23
  #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
24
+ pipe.tokenizer.model_max_length=512
25
  pipe.to("cuda")
26
 
27
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: