jjz5463 commited on
Commit
48fe872
·
1 Parent(s): 3bfad84

change back to float 16

Browse files
Files changed (1) hide show
  1. baseline_utils.py +2 -2
baseline_utils.py CHANGED
@@ -77,8 +77,8 @@ def analyze_writer_image(image_path, api_key):
77
  def generate_comic_book(diary_text, writer_description, num_pages=4):
78
  pipe = AutoPipelineForText2Image.from_pretrained(
79
  "stabilityai/sdxl-turbo",
80
- torch_dtype=torch.float32,
81
- variant="fp32",
82
  cache_dir="./SDXL-Turbo"
83
  )
84
 
 
77
  def generate_comic_book(diary_text, writer_description, num_pages=4):
78
  pipe = AutoPipelineForText2Image.from_pretrained(
79
  "stabilityai/sdxl-turbo",
80
+ torch_dtype=torch.float16,
81
+ variant="fp16",
82
  cache_dir="./SDXL-Turbo"
83
  )
84