Spaces:

baulab
/

ConceptSliders

Runtime error

RohitGandikota commited on Nov 28, 2023

Commit

69ec7a4

1 Parent(s): 461b806

Update trainscripts/textsliders/data/config-xl.yaml

Files changed (1) hide show

trainscripts/textsliders/data/config-xl.yaml CHANGED Viewed

@@ -9,7 +9,7 @@ network:
   alpha: 1.0
   training_method: "xattn"
 train:
-  precision: "bfloat16"
   noise_scheduler: "ddim" # or "ddpm", "lms", "euler_a"
   iterations: 1000
   lr: 0.0002

   alpha: 1.0
   training_method: "xattn"
 train:
+  precision: "fp16"
   noise_scheduler: "ddim" # or "ddpm", "lms", "euler_a"
   iterations: 1000
   lr: 0.0002