RohitGandikota commited on
Commit
0330657
Β·
1 Parent(s): 0130cd3

fixing cuda issues

Browse files
trainscripts/textsliders/data/config-xl.yaml CHANGED
@@ -7,7 +7,7 @@ network:
7
  type: "c3lier" # or "c3lier" or "lierla"
8
  rank: 4
9
  alpha: 1.0
10
- training_method: "noxattn"
11
  train:
12
  precision: "bfloat16"
13
  noise_scheduler: "ddim" # or "ddpm", "lms", "euler_a"
@@ -25,4 +25,4 @@ logging:
25
  use_wandb: false
26
  verbose: false
27
  other:
28
- use_xformers: false
 
7
  type: "c3lier" # or "c3lier" or "lierla"
8
  rank: 4
9
  alpha: 1.0
10
+ training_method: "xattn"
11
  train:
12
  precision: "bfloat16"
13
  noise_scheduler: "ddim" # or "ddpm", "lms", "euler_a"
 
25
  use_wandb: false
26
  verbose: false
27
  other:
28
+ use_xformers: false