jjz5463 commited on
Commit
8127d0e
1 Parent(s): f9ea10e

comment out torch_dtype=torch.float16

Browse files
Experiments/Baseline/baseline_utils.py CHANGED
@@ -77,7 +77,7 @@ 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.float16,
81
  variant="fp16",
82
  cache_dir="./SDXL-Turbo"
83
  )
 
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
  )